jax
1b1e9f71 - Internal refactor: move TPU lowering rules out of jax/_src/lax

Commit

235 days ago

Internal refactor: move TPU lowering rules out of jax/_src/lax

References

#29420 - Internal refactor: move TPU lowering rules out of jax/_src/lax

Author

jakevdp

jakevdp

Parents

Loading