JAX 0.4.4 introduced a bug that was caught by a Dynamics test (discussed here). This was temporarily fixed by #189 , which bypassed the new code. As of JAX 0.4.7, this bypass can no longer be used. This PR sets the upper bound to 0.4.6, the latest version that the bypass works.
To get the tests functioning, this PR also includes corrections to the DynamicsBackend tutorial.
Summary
Closes #205.
JAX 0.4.4 introduced a bug that was caught by a Dynamics test (discussed here). This was temporarily fixed by #189 , which bypassed the new code. As of JAX 0.4.7, this bypass can no longer be used. This PR sets the upper bound to 0.4.6, the latest version that the bypass works.
To get the tests functioning, this PR also includes corrections to the
DynamicsBackendtutorial.Details and comments