https://github.com/google/jax
History
Tip revision: 9424ff70dee59c23e81db4cee4e2660232363449 authored by Sharad Vikram on 06 March 2024, 17:46:25 UTC
Change `mosaic_params=...` to `compiler_params=dict(mosaic=...)` to use new convention for passing in compiler parameters.
Tip revision: 9424ff7
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-- 16.5 KB
README.md -rw-r--r-- 4.8 KB
Wave_Equation.ipynb -rw-r--r-- 15.4 KB

README.md

back to top