A Jax based neural network library for research
Project description
fastax
Overview
fastax is a Deep Learning Library on top of Google's JAX that tries to facilitate doing research with it by providing many a combination of JAX's versatility with implementations of the most used neural networks layers and utilities.
More concrete information and naming
fastax is both a combination of fastai and JAX and fastai and stax:
- It is built on top of JAX and most of its original code comes from its Neural Network mini-library stax
- One of the long term goals of this library is implementing Jeremy Howard's fastai course "Deep Learning From the Foundations"
For more information visit https://github.com/joaogui1/fastax
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
fastax-0.1.0.tar.gz
(12.5 kB
view hashes)
Built Distribution
fastax-0.1.0-py3-none-any.whl
(14.5 kB
view hashes)
