[fix] Replace jax.experimental.host_callback with jax.pure_callback#668
Closed
Routhleck wants to merge 14 commits intomasterfrom fix_pure_callback
+140-129
Commits
Commits on May 6, 2024
Commits on May 8, 2024
- committed
- committed
Commits on May 10, 2024
- committed
- committed
- committed
- authored
- committed
- committed
Commits on May 12, 2024
- committed