Commits

History for

Commits on Sep 26, 2023

[PJRT C API] Register custom callback for xla_python_gpu_callback in plugin module.

jyingl3
authored and
jax authors
committedSep 26, 2023

Commits on Sep 25, 2023

[PJRT C API] Implement framework side change for registering a custom call.

jyingl3
authored and
jax authors
committedSep 25, 2023

Commits on Sep 13, 2023

[PJRT C API] Set up jax xla cuda package.

jyingl3
authored and
jax authors
committedSep 13, 2023

Loading