jax
jax copied to clipboard
Add functionality for "pure" callbacks
Add functionality for "pure" callbacks
Also avoids using CPP dispatch path when host callbacks are involved