Sergei Lebedev comments

Results 228 comments of


                                            Sergei Lebedev

[Pallas TPU] Add lowering for `lax.erf_inv` 64 bit

We run some tests on x64, but not all, yeah.

`jax.nn.dot_product_attention` does not respect `key_value_seq_lengths`

@kaixih PTAL.

Support dlpack pinned host memory device type (kDLCPUPinned / kDLCUDAHost)

We are ready to support this on our side, but it looks like PyTorch does not currently use kDLCUDAHost, see pytorch/pytorch#136250.

Navigating the stack in a coredump resulting from a stack overflow is slow

> [...] how deep into the stack are you going? Pretty deep, millions of frames. > Yes, making a stack trace takes linear time (and, currently, memory) in the depth...

Navigating the stack in a coredump resulting from a stack overflow is slow

Making a huge stack trace sadly won't work for my use case, since I need to print local variables/arguments in the bottom frames.

Vendor wcwidth using python's unicodedata

I suspect you might need to keep the MIT license for that file, since the implementation follows the one in pytest.

Vendor wcwidth using python's unicodedata

Can you rebase, please?

[RFC] Latest status of draft

@mikeshardmind thank you for working on this! Could you upload the draft on Google Docs or make it a PR so that it is easier for people to ask questions/leave...