/machine-learning-with-jax

A practice for building machine learning models by jax

Primary LanguagePythonApache License 2.0Apache-2.0

machine-learning-with-jax

A practice for building machine learning models by jax