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