3 ms·

This isn't really comparable to JAX afaict - JAX offers only masked dense computation, this is first-class ragged support with no mention of autodiff or jit (ou

by nmca 5y ago

This isn't really comparable to JAX afaict - JAX offers only masked dense computation, this is first-class ragged support with no mention of autodiff or jit (outside of numba)

hyperbovine 5y ago
There is definitely a comparison to be made, because JAX gets a big speed gain by embracing that limitation.