Last released Jul 28, 2026
Minimal JAX interpreter layer for threading custom context through traced computations
Last released Jul 23, 2026
A tiny config loader and CLI framework built around Pydantic models
Last released Jul 1, 2026
A toolkit for building high-performance molecular simulations on JAX.
Effortlessly transform PyTorch functions to JAX
Supported by