[fix] Replace jax.experimental.host_callback
with jax.pure_callback
#1242
Annotations
2 errors
Set up Python 3.10
[notice] A new release of pip is available: 23.0.1 -> 24.0
[notice] To update, run: python3.10 -m pip install --upgrade pip
|
Test with pytest
Process completed with exit code 1.
|
Loading