Closed FFroehlich closed 3 months ago
host_callback is deprecated (see https://jax.readthedocs.io/en/latest/jax.experimental.host_callback.html), use pure_callback instead (see https://jax.readthedocs.io/en/latest/notebooks/external_callbacks.html)
host_callback
pure_callback
host_callback
is deprecated (see https://jax.readthedocs.io/en/latest/jax.experimental.host_callback.html), usepure_callback
instead (see https://jax.readthedocs.io/en/latest/notebooks/external_callbacks.html)