Tristan Bilot

icon indicating a website link https://tristanbilot.github.io

icon company Université Paris-Saclay icon location Paris, France icon location Ph.D. student | GNNs for Cybersecurity.

Results 24 comments of Tristan Bilot

Alternative to nonzero()

Hmm I see, one workaround would require running a first kernel to count the number of values that verify the condition, then allocate the appropriate memory and then running the...

Alternative to nonzero()

> * In Jax you can't JIT through ops like `nonzero` unless you explicitly pass a `size` parameter. > * I haven't looked at PyTorch MPS but my guess is...

PyTorch implementation

Any updates?

Regular benchs ?

It would be great to have updated benchmarks using the latest version of MLX. Everyone is free to open a PR to integrate results with more recent versions!