mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-27 04:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
18,532 Commits 508 Branches 263 Tags
Commit Graph

11 Commits

Author SHA1 Message Date
8bitmp3
5eebb91e1e Upgrade JAX pytrees doc 2023-12-07 19:18:04 +00:00
jax authors
dc469c3851 Merge pull request #18676 from 8bitmp3:jax-docs-autodiff-101-201 
PiperOrigin-RevId: 588172128
 2023-12-05 13:29:10 -08:00
8bitmp3
46bad2bd62 Upgrade JAX Autodiff 101 2023-12-05 00:00:13 +00:00
8bitmp3
b1a8bc6a83 Upgrade JAX Installation doc 2023-12-04 21:42:13 +00:00
Jake VanderPlas
83cb3369d2 JAX tutorials: quickstart 2023-11-28 08:28:18 -08:00
jax authors
961ba3cd42 Merge pull request #18631 from jakevdp:tutorials-random 
PiperOrigin-RevId: 584612036
 2023-11-22 06:42:14 -08:00
Jake VanderPlas
52adb1c6a5 JAX tutorials: add intro to jit compilation 2023-11-21 14:27:44 -08:00
Jake VanderPlas
cacadf43c0 JAX tutorials: pseudorandom numbers 2023-11-21 14:26:06 -08:00
Jake VanderPlas
29a2e8a362 JAX tutorials: add automatic vectorization 2023-11-21 10:39:02 -08:00
Jake VanderPlas
66a3eff28f JAX tutorials: external callbacks 2023-11-20 17:04:05 -08:00
Jake VanderPlas
7456921055 Move docs/tutorial to docs/tutorials 2023-11-17 10:24:56 -08:00