https://github.com/google/jax
History
Tip revision: d4943ff4f811c2216ed2d90aad0e84ca64078b7d authored by Tomás Longeri on 04 October 2023, 07:35:23 UTC
[Mosaic] apply_vector_layout C++ rewrite: Enable tpu_custom_call_test for C++ (in addition to Python)
Tip revision: d4943ff
File Mode Size
images
JAX_NeurIPS_2020_demo.ipynb -rw-r--r-- 10.5 KB
JAX_demo.ipynb -rw-r--r-- 19.1 KB
Lorentz_ODE_Solver.ipynb -rw-r--r-- 14.0 KB
Pmap_Cookbook.ipynb -rw-r--r-- 17.4 KB
README.md -rw-r--r-- 4.8 KB
Wave_Equation.ipynb -rw-r--r-- 15.4 KB

README.md

back to top