23 Commits

Author	SHA1	Message	Date
Jake VanderPlas	93500a8477	lint: fix jupytext version	2024-01-09 15:28:17 -08:00
jax authors	77cb0f141c	Merge pull request #19147 from 8bitmp3:jax-docs-jaxprs ... PiperOrigin-RevId: 597057971	2024-01-09 15:21:43 -08:00
Jake VanderPlas	10eae3f93a	CI: update jupytext version	2024-01-09 14:34:21 -08:00
8bitmp3	66a845e0c6	Upgrade JAX internals 301 jaxpr language tutorial	2024-01-09 21:53:22 +00:00
jax authors	04f2c91399	Merge pull request #19092 from 8bitmp3:jax-docs-gradient-checkpointing ... PiperOrigin-RevId: 597034325	2024-01-09 13:50:13 -08:00
8bitmp3	efeb20e380	Upgrade JAX Gradient Checkpointing doc	2024-01-08 23:36:21 +00:00
8bitmp3	6afb83a463	Upgrade JAX Advanced Autodiff 201	2023-12-20 23:58:49 +00:00
8bitmp3	7edc80d635	Update thinking-in-jax working-with-pytrees	2023-12-13 22:57:46 +00:00
8bitmp3	a4003bf0ae	Upgrade How to think in JAX	2023-12-13 21:39:17 +00:00
jax authors	0281596ec9	Merge pull request #18869 from 8bitmp3:small-fix-jax-pytrees ... PiperOrigin-RevId: 590680938	2023-12-13 12:40:06 -08:00
8bitmp3	7bb4b0fab4	Update working-with-pytrees.md	2023-12-13 00:40:42 +00:00
8bitmp3	14407b9a06	Upgrade JAX debugging doc	2023-12-13 00:19:41 +00:00
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