Add JAX 0.3.x compatibility. Useful for running experiments on IPUs. #126
tests.yaml
on: pull_request
lint_and_typecheck
17s
unit_tests
38s