Ian Horn
Ian Horn
### Description [JAX](https://github.com/google/jax) includes a numpy compatible `jax.numpy` module which has a bunch of nice features (automatic differentiation, jit compilation, vectorized mapping, GPU runtime, js export). They've taken great pains...
Working with jagged tensors is a real pain, especially when you have to translate back and forth from a fixed layout mentality to different jagged mentalities. Einops notation makes this...
## 🐞Describing the bug Using a mix of static and range input shapes with a stateful model fails. While trying to make a minimal repro, I encountered some other strange...