Skip to content

[fix] Replace jax.experimental.host_callback with jax.pure_callback #1252

[fix] Replace jax.experimental.host_callback with jax.pure_callback

[fix] Replace jax.experimental.host_callback with jax.pure_callback #1252

test_linux (3.11)

succeeded May 14, 2024 in 36m 48s