search

google / aqt

Apache License 2.0
247 stars 25 forks source link

Remove usage of xla_computation and replace with jax.jit(f).lower(*args).compile_ir() #646

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Remove usage of xla_computation and replace with jax.jit(f).lower(*args).compile_ir()