8bitmp3 a295b7859f Upgrade JAX primitives doc

2024-01-24 00:47:46 +00:00

..

advanced-autodiff.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

advanced-compilation.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

advanced-debugging.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

automatic-differentiation.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

automatic-vectorization.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

debugging.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

external-callbacks.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

gradient-checkpointing.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

index.rst

Upgrade JAX primitives doc

2024-01-24 00:47:46 +00:00

installation.md

DOC: add a note about installing older GPU wheels

2024-01-12 11:12:39 -08:00

jax-as-accelerated-numpy.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

jax-primitives.md

Upgrade JAX primitives doc

2024-01-24 00:47:46 +00:00

jaxpr.md

lint: fix jupytext version

2024-01-09 15:28:17 -08:00

jit-compilation.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

parallelism.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

profiling-and-performance.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

quickstart.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

random-numbers.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

simple-neural-network.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

single-host-sharding.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

stateful-computations.md

Move docs/tutorial to docs/tutorials

2023-11-17 10:24:56 -08:00

thinking-in-jax.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00

working-with-pytrees.md

CI: update jupytext version

2024-01-09 14:34:21 -08:00