[bug] Fixing compatibility issues with jax
#1304
CI-models.yml
on: pull_request
Matrix: test_linux
Matrix: test_macos
Matrix: test_windows