Pytorch-like modules in Jax
Project description
Flarejax
Pytorch-like modules in Jax
TODO
- Wrap module init for saving attempting to save args and kwargs in json?
- Sequential and Linear
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
flarejax-0.1.0.tar.gz
(3.9 kB
view hashes)