flax-pilot

PyPI page
Home page
Author: Nithish M
Summary: A Simplistic trainer for Flax
Latest version: 0.2.5
Required dependencies: flax | jax | mergedeep | numpy | optax | orbax-checkpoint | tqdm

Downloads last day: 4
Downloads last week: 25
Downloads last month: 56