PyPI
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Apache-2.021. Okt. 2026 · Metriken 2.10.0
Alle Tags
Katalog-Tag
#flax
Alle Repositories im öffentlichen Register, die dieses Tag tragen — aus ihren GitHub-Topics oder den von ihren Paket-Registries veröffentlichten Schlagwörtern. Die Gesundheit wird nach derselben versionierten Methodik gemessen wie im übrigen Register.
6 Einträge
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. Okt. 2026 · Metriken 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 · Metriken 2.10.0
NuGet · npm
outfox/fennecs
...the little C# ECS that loves you back! 🦊
MIT25. Sept. 2026 · Metriken 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 · Metriken 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 · Metriken 2.10.0