Create JAX's custom call shim for GPUs

Open vmurali opened this issue 2 years ago • 0 comments

This should be feature complete - support for arbitrary types and aritie of inputs and outputs.

Apr 19 '23 18:04 vmurali