7 projects
xtrax
High-performance composable JAX library
proxide
Protein I/O for JAX with high-performance Rust backend
bathos
Local-first, zero-server experiment tracking for researchers
aminx
Aminx: A functional interface for ProteinMPNN
jaxlint
CLI and library for JAX-oriented static checks, docstring quality, and optional HLO analysis
psax
JAX implementation of Protein Strain Analysis (per-site weighted finite strain, PSA)
expaloma
JAX/Equinox port of espaloma-charge partial charge inference (bundled v0.0.8 weights)
