jax

d5c7ccc7 - [xla:python] Add support for registering custom call targets for all XLA execution stages and for XLA FFI traits