/ml_basics

Building and Training Basic Barebones ML Models in JAX.

Primary LanguagePythonMIT LicenseMIT

ML Basics

A repository containing barebones ML implementations in JAX.

How to run:

All you need is JAX. Follow the steps below to clone the repository, create a virtual environment, and install JAX for the CPU.

git clone git@github.com:jeh15/ml_basics.git
cd sandbox
python3 -m venv env
source env/bin/activate
pip install --upgrade pip
pip install -U "jax[cpu]"