Add JAX 0.3.x compatibility. Useful for running experiments on IPUs. #127
tests.yaml
on: pull_request
lint_and_typecheck
29s
unit_tests
57s