bax

A flexible trainer interface for Jax and Haiku.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install bax

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
0.1.12 2022-05-30  
0.1.11 2022-03-24  
0.1.10 2022-03-09  
0.1.9 2022-03-05  
0.1.8 2022-02-16  
0.1.7 2022-02-16  

Issues with this package?

Page last updated 2025-07-18 04:06:52 UTC