Facilitates machine learning development for JAX.
Project description
EasyJax
work-in-progress.
EasyJax is a python package that facilitates machine learning development for JAX. It does that by providing:
- A high-level API for machine learning workflows in JAX (specifically a trainer, experiment parent class).
- Several machine learning specific utilities for working with JAX (e.g.,
ml.update_step
).
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
easyjax-0.1.3.tar.gz
(2.7 kB
view hashes)