This repository was archived by the owner on Nov 6, 2024. It is now read-only.
This repository was archived by the owner on Nov 6, 2024. It is now read-only.
Jax errors when trying to run tests and examples #17
Description
Receiving errors like this:
ValueError: Jitted function has static_argnums=(0, 6), but only accepts 6 positional arguments.
in lots of different places. I'm using jax=0.4.30