jax
39f7db68 - Change jaxlib/rocm/BUILD nanobind_extension to pybind (#226)

Commit

1 year ago

Change jaxlib/rocm/BUILD nanobind_extension to pybind (#226)

References

#226 - Change jaxlib/rocm/BUILD nanobind_extension to pybind_extension

#266 - Update XLA Commit hash

#305 - Add support for ROCm wheel based install

Author

Ruturaj4

Ruturaj4

Parents

Loading