Practice Questions of JAX

This is a practice project for demonstrating your familiarity of JAX.

It contains following small projects:

• 00_data_preprocessor.ipynb

How to Get Started

1. Clone this repo:

git clone https://github.com/BirkhoffG/jax-practice.git

2. Install all the dependencies:

cd jax-practice
pip install -e ".[dev]"

3. Start working on these small projects by editing the code in the jupyter notebook.

4. Once it is done, create a pull request.

Some Helpful Tips

• We are using nbdev for our library development. This repo is also based on nbdev. It will be helpful to know what these comments are: nbdev_export, nbdev_clean, nbdev_test.
  • This tutorial is a good start in learning nbdev: https://nbdev.fast.ai/tutorials/tutorial.html
  • Learn how to use nbdev in jax-relax: https://birkhoffg.github.io/jax-relax/tutorials/contribution.html