mlxu (pypi): Machine learning experiment utils.
Keywords: jax|machine-learning|python|wandb