PyPI page
Home page
Author:
kasinadhsarma
License:
MIT
Summary:
A JAX-based neural network library surpassing Google DeepMind's Haiku and Optax
Latest version:
0.2.0
Required dependencies:
chex
|
dm-haiku
|
fairscale
|
flax
|
gym
|
jax
|
jaxlib
|
moviepy
|
numpy
|
optax
|
pillow
|
scipy
|
stable-baselines3
|
sympy
|
tensorflow
|
torch
|
transformers
|
whisper
Optional dependencies:
flake8
|
pytest
Downloads last day:
0
Downloads last week:
5
Downloads last month:
19