mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/lax
History
Tianjian Lu c5f73b3d8e [JAX] Added jax.lax.linalg.qdwh. 
PiperOrigin-RevId: 406453671
2021-10-29 14:45:06 -07:00
..
__init__.py
[JAX] Port lax translation rules to updated XLA translation rule API.
2021-10-18 18:07:29 -07:00
linalg.py
[JAX] Added jax.lax.linalg.qdwh.
2021-10-29 14:45:06 -07:00