https://github.com/google/jax
History
Tip revision: a1c03644394a1cb92fd70e64323968a8b5616c5d authored by Peter Hawkins on 06 June 2024, 18:15:31 UTC
Fix a bug where treedef.flatten_up_to(...) was overly permissive for None treedefs.
Tip revision: a1c0364
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.4 KB
README.md -rw-r--r-- 4.3 KB
Wave_Equation.ipynb -rw-r--r-- 15.4 KB

README.md

back to top