Bijections & normalizing flows with JAX/NNX
-
Updated
Mar 11, 2026 - Python
#
nnx
Here are 8 public repositories matching this topic...
Reasoning-from-Zero using gemma.JAX.nnx on TPUs
-
Updated
Jan 20, 2026 - Python
Model zoo and experimental features of machine learning interatomic potentials.
molecular-dynamics molecular-simulation jax machine-learning-force-field ai-for-science machine-learning-interatomic-potential nnx mlff mlip
-
Updated
Mar 13, 2026 - Python
Improve this page
Add a description, image, and links to the nnx topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the nnx topic, visit your repo's landing page and select "manage topics."