For our open-source software, visit cusp-ai-oss.
Major releases
- kUPS — A toolkit for building high-performance molecular simulations on JAX
- tojax — Seamless translation of PyTorch models and functions to JAX
Additional tooling
- tonno — Autotuning decorator for JAX/Pallas kernels with on-device benchmarking and caching
- slub — Minimal JAX interpreter layer for threading a custom context through traced computations
- nanoargs — A tiny config loader and CLI framework built around Pydantic models
- widom — Widom insertion for computing adsorption properties of porous materials