https://github.com/google/jax
History
Tip revision: 9fffa8644ab7e3d2693b362f92bdcf6d6b2e55c9 authored by Matthew Johnson on 09 February 2021, 04:24:19 UTC
add pargmin_p / pargmax_p primitives
Tip revision: 9fffa86
File Mode Size
images
JAX_NeurIPS_2020_demo.ipynb -rw-r--r-- 10.5 KB
JAX_demo.ipynb -rw-r--r-- 19.4 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-- 5.8 KB
Wave_Equation.ipynb -rw-r--r-- 15.6 KB

README.md

back to top