https://github.com/google/jax
History
Tip revision: 5c4e88d8799d8e24ca5bfba5a84c177a66b21f92 authored by Yash Katariya on 09 August 2023, 02:00:31 UTC
[Memories] Allow device_put outside jax.jit to work with numpy arrays, numpy scalars, python scalars and python ints with memory kinds
Tip revision: 5c4e88d
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