[fix] Replace jax.experimental.host_callback with jax.pure_callback
#1151
Annotations
1 error
|
Set up Python 3.9
[notice] A new release of pip is available: 22.0.4 -> 24.0
[notice] To update, run: python3.9 -m pip install --upgrade pip
|
Loading