https://github.com/google/jax
History
Tip revision: 57746df4289d2f62daa8a1fde6daa0d81f1403cc authored by Tianjian Lu on 26 April 2022, 21:53:53 UTC
[linalg] Add tpu svd lowering rule.
Tip revision: 57746df
File Mode Size
images
JAX_NeurIPS_2020_demo.ipynb -rw-r--r-- 10.5 KB
JAX_demo.ipynb -rw-r--r-- 19.5 KB
Lorentz_ODE_Solver.ipynb -rw-r--r-- 14.7 KB
Pmap_Cookbook.ipynb -rw-r--r-- 17.9 KB
README.md -rw-r--r-- 4.8 KB
Wave_Equation.ipynb -rw-r--r-- 15.5 KB

README.md

back to top