Dan Foreman-Mackey
Results
94
issues of
Dan Foreman-Mackey
In https://github.com/jax-ml/jax/pull/23261, the lowering of `lax.tan` was updated to support the native StableHLO lowering, but jaxlib v0.4.33 didn't include the necessary re-writes to support this lowering via the PJRT plugin...
pull ready
We have an end-to-end example demonstrating the use of CUDA for FFI calls in `docs/cuda_custom_call`. Because we didn't previously have GPU runners this was tested using bazel using an internal...