ninjax icon indicating copy to clipboard operation
ninjax copied to clipboard

General Modules for JAX

Results 5 ninjax issues
Sort by recently updated
recently updated
newest added

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...