Show HN: NumPy+Jax Except with Named Axes

If this idea sounds interesting, you might want to look at xarray for a more established project: https://github.com/pydata/xarray

It's focused on indexing and lacks the batching idea, though. Combining named axes and vmap is a great move!

But tbh I'm turned off by the repo not being a Python package more than I'd like to admit. Reading

> Download it and put it in your working directory.

when skimming a project makes me think it's immature. Which definitely is not the case here!