search

iree-org / iree-jax

Apache License 2.0
48 stars 19 forks source link

JAX tracing op #60

Open jpienaar opened 1 year ago

jpienaar commented 1 year ago

Op exposed up to JAX that folks could insert to use IREE tracing.