Last released Apr 23, 2026
A toolkit for building high-performance molecular simulations on JAX.
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
