Use PJRT-based custom call registration to register standard JAX custom calls (e.g. Eigh)

Open vmurali opened this issue 2 years ago • 0 comments

For CPUs, register these custom calls. For GPUs, register these custom calls.

Apr 19 '23 18:04 vmurali