3 ms·
It would be nice to see some comparison to JAX. Is it mostly that arrays in Awkward are, sort of, unstructured? JIT-ed, ADed JAX on JSON-like structures, on GP
by rich_sasha 5y ago
It would be nice to see some comparison to JAX.
Is it mostly that arrays in Awkward are, sort of, unstructured? JIT-ed, ADed JAX on JSON-like structures, on GPU?
- ivirshup 5y agoIt’s a ragged array that you can work with in python code, but also Jax and numba code.
- nmca 5y agoThis 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 agoThere is definitely a comparison to be made, because JAX gets a big speed gain by embracing that limitation.