PyPI Stats

Search

All packages
Top packages

Track packages

trainax


PyPI page
Home page
Author: Felix Koehler
Summary: Training methodologies for Autoregressive Neural Emulators in JAX built on top of Equinox & Optax.
Latest version: 0.0.2
Required dependencies: equinox | jax | jaxtyping | optax | tqdm | typing-extensions

Downloads last day: 19
Downloads last week: 2,884
Downloads last month: 12,443