jax_demo

This repo is for those who have experience in deep learning with torch, and want to learn a lesser known but still powerful library JAX

Requirements

pip install jax or pip install -U "jax[cuda12]" if using a CUDA GPU

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
README.md		README.md
as_numpy.ipynb		as_numpy.ipynb
bayes.ipynb		bayes.ipynb
nn_features.ipynb		nn_features.ipynb