Skip to content

JAX Tutorial for Autonomous Learning Group Retreat

Notifications You must be signed in to change notification settings

jimimvp/jax-tutorial

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

The JAX Tutorial

Follow these installation steps.

0. Python

We'll use Python 3.7.6 One option to install it is using https://github.com/pyenv/pyenv-installer or install it locally however you see fit.

1. Dependencies

Option 1 (virtual envs are highly recommended though):

pip3.7 install -r requirements.txt

Option 2: for the fancy people is using https://python-poetry.org/. Once you install poetry, you can run this in the project root:

poetry install 
poetry shell

2. Jupyter

If everything went well with the installation, you should be able to run this in the terminal, then open up the notebook in the browser (run in root folder of project):

python3.7 -m ipykernel install --user --name jax-tutorial
jupyter notebook

Next steps, open up tutorial.ipynb in jupyter notebook, select jax-tutorial and follow along.

About

JAX Tutorial for Autonomous Learning Group Retreat

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published