jejjohnson / eqx-trainer

A trainer module for the equinox (JAX) framework.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A Simple Trainer for neural networks with Equinox


Installation

This package isn't pip-worthy (yet) but here are a few options for installation.

Option I: Use the pip install option (locally)

https://github.com/jejjohnson/eqx-trainer.git
cd eqx-trainer
pip install -e .[dev,all]

Option II: Install it from pip directly.

pip install "git+https://github.com/jejjohnson/eqx-trainer.git"

Inspiration

About

A trainer module for the equinox (JAX) framework.

License:MIT License


Languages

Language:Python 72.8%Language:Jupyter Notebook 25.4%Language:Makefile 1.2%Language:Dockerfile 0.7%