Giters

jeh15 / ml_basics

Building and Training Basic Barebones ML Models in JAX.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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]"

About

Building and Training Basic Barebones ML Models in JAX.

License:MIT License

Languages

Language:Python 100.0%