astax

A Jax based neural network library for research


License
Apache-2.0
Install
pip install astax==0.1.3