PyPI
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Apache-2.021 oct 2026 · métricas 2.10.0
Todas las etiquetas
Etiqueta del catálogo
#flax
Todos los repositorios del registro público que llevan esta etiqueta, procedente de sus topics de GitHub o de las palabras clave que publican sus registros de paquetes. La salud se mide con la misma metodología versionada que el resto del registro.
6 registros
PyPI
avitai/artifex
Generative modeling for JAX/Flax NNX: VAEs, GANs, DDPM/DDIM/score/DiT/latent diffusion, normalizing flows, energy-based, autoregressive and geometric (SE(3), protein, point cloud, mesh) models; image, text, audio, molecular, protein, tabular and time-series modalities; a shared Trainer with callbacks and extensions; evaluation through calibrax.
MIT21 oct 2026 · métricas 2.10.0
PyPI
avitai/substrax
JAX/Flax NNX training infrastructure: device meshes and SPMD sharding, an Orbax checkpoint store, early stopping and callbacks, W&B and MLflow logging.
MIT26 sept 2026 · métricas 2.10.0
NuGet · npm
outfox/fennecs
...the little C# ECS that loves you back! 🦊
MIT25 sept 2026 · métricas 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.
MIT9 sept 2026 · métricas 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.
MIT9 sept 2026 · métricas 2.10.0