Add JAX 0.3.x compatibility. Useful for running experiments on IPUs. #128
tests.yaml
on: pull_request
lint_and_typecheck
28s
unit_tests
37s