CuspAI

CuspAI Open Source Software

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