Skip to content

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

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

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

Annotations

2 errors

test_macos (3.10)

failed May 12, 2024 in 17m 53s