Ian Horn

Results 3 issues of 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...

Feature

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...

feature suggestion

## 🐞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...

bug
triaged