iree
iree copied to clipboard
Create JAX's custom call shim for GPUs
This should be feature complete - support for arbitrary types and aritie of inputs and outputs.