Einshape

github.comtool

DSL-based reshaping library for JAX and other frameworks.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   imax github.com

   tool category #jax link ↗

   Image augmentations and transformations.

   
2. ▲
   JaxDF github.com

   tool category #jax link ↗

   Framework for differentiable simulators with arbitrary discretizations.

   
3. ▲
   jaxns github.com

   tool category #jax link ↗

   Nested sampling in JAX.

   
4. ▲
   QDax github.com

   tool category #jax link ↗

   Quality Diversity optimization in Jax.

   
5. ▲
   dendritex github.com

   tool category #jax link ↗

   Dendritic Modeling in JAX.

   
6. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   

JAX › Libraries > New Libraries: “DSL-based reshaping library for JAX and other frameworks.”