PyPI
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Apache-2.0Aug 29, 2026 · metrics 2.10.0
All tags
Catalogue tag
#flax
Every repository in the public record carrying this tag — from its GitHub topics or the keywords its package registries publish. Health is measured under the same versioned methodology as the rest of the record.
4 records
PyPI
avitai/artifex
A research-focused modular generative modeling library built on JAX/Flax NNX
MITAug 29, 2026 · metrics 2.10.0
PyPI
elkins-lab/resonance-flow
JAX-native differentiable protein folding framework integrating experimental NMR constraints (RDCs) and biophysical "self-correction." Several Jupyter Notebooks visualize the concepts.
MITJul 17, 2026 · metrics 2.10.0
PyPI
elkins-lab/diff-ensemble
Differentiable VAE framework for predicting protein structural ensembles (IDPs) consistent with SAXS and NMR data. Built on JAX/Flax.
MITJul 17, 2026 · metrics 2.10.0