conda create -n getting-started-with-jax python=3.10 ipykernel jupyter
conda activate getting-started-with-jax
pip install jax
# Optional (Apple Silicon)
pip install jax-metal-
Notifications
You must be signed in to change notification settings - Fork 0
berianjames/getting-started-with-jax-for-ml
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Getting Started with JAX for Machine Learning (D3A tutorial)
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published