Sai-Suraj-27
29def4eefa
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
Peter Hawkins
478cfa9944
Add an upper bound on JAX's CUDNN version constraint.
...
Major releases of CUDNN break ABI compatibility, so we cannot allow new major versions.
PiperOrigin-RevId: 620030416
2024-03-28 13:00:36 -07:00
jax authors
0be07e6aec
Remove support for CUDA 11.
...
Pin minimal required versions for CUDA to 12.1.
Reverts 910a31d7b7510e3375718ab1ea0d38df7bd2c0d5
PiperOrigin-RevId: 618911489
2024-03-25 11:46:39 -07:00
jax authors
910a31d7b7
Reverts bed4f65438a62777ed100ecec2b0eb3f7cf87a0e
...
PiperOrigin-RevId: 618249855
2024-03-22 12:10:53 -07:00
jax authors
bed4f65438
Remove support for CUDA 11.
...
Pin minimal required versions for CUDA to 12.1.
PiperOrigin-RevId: 618195554
2024-03-22 09:05:39 -07:00
Jake VanderPlas
d1235fac8e
DOC: add key concepts doc
...
This will replace the new content in thinking-in-jax within the new tutorial flow.
2024-03-20 18:30:55 -07:00
Jake VanderPlas
d6c07bdf51
DOC: read-through and edit the new jax tutorials
2024-03-20 18:18:08 -07:00
rajasekharporeddy
61c64c10f8
Fixed Several Typos
...
Fixed Typos in JEP doc files
Revert "Fixed Typos in JEP doc files"
This reverts commit c2a16950e0fc1b32971168501d183991e2394b5d.
revert two changes
reverted one change in advanced-autodiff
revert one change in parallelism
sync notebooks
2024-03-12 00:37:46 +05:30
8bitmp3
a295b7859f
Upgrade JAX primitives doc
2024-01-24 00:47:46 +00:00
Jake VanderPlas
c785eeb4e5
DOC: add a note about installing older GPU wheels
2024-01-12 11:12:39 -08:00
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
