GitHub - gordicaleksa/get-started-with-JAX at dcf744b29c9256bd7e2818ca0bcbaec28b7fdb0a

gordicaleksa / get-started-with-JAX Public

Notifications You must be signed in to change notification settings
Fork 100
Star 677

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

www.youtube.com/c/theaiepiphany/

MIT license

677 stars 100 forks Branches Tags Activity

Star

Notifications

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
.gitignore		.gitignore
LICENCE		LICENCE
README.md		README.md
Tutorial_1_JAX_basics.ipynb		Tutorial_1_JAX_basics.ipynb

Repository files navigation

Get started with JAX!

Citation

If you find this code useful, please cite the following:

@misc{Gordić2021GetStartedWithJAX,
  author = {Gordić, Aleksa},
  title = {Get started with JAX},
  year = {2021},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/gordicaleksa/get-started-with-JAX}},
}