ninjax
ninjax copied to clipboard
General Modules for JAX
Hi, I tried to run the script you provided with `ninjax==2.3.3`, the script is: ```python import flax import jax import jax.numpy as jnp import ninjax as nj import optax #...
Hi Danijar, I'm working with ninjax==0.9.0 since it is used in your DreamerV3 implementation. I found that your doc in https://pypi.org/project/ninjax/0.9.0/ has some typo error and will lead to the...
Add connectivity check to address #2
Hi, Thank you for the nice framework, I want to run your file "quickstart.py" and got the following compiler error: > File "/nfs/hpc/share/nguyejoe/dynalang/ninjax_test/quickstart.py", line 8, in import ninjax as nj...
I might take a stab at these myself; but here are some ideas for examples that might showcase the strengths of ninjax - batch normalization; known to feel tricky to...