PyPI page
Home page
Author:
None
License:
Apache-2.0
Summary:
A simple and useable UED library in Jax
Latest version:
0.0.1
Required dependencies:
chex
|
flax
|
gymnax
|
jax
|
numpy
Optional dependencies:
distrax
|
imageio
|
moviepy
|
optax
|
orbax
|
orbax-checkpoint
|
pillow
|
wandb
Downloads last day:
5
Downloads last week:
72
Downloads last month:
250