jax authors 0281596ec9 Merge pull request #18869 from 8bitmp3:small-fix-jax-pytrees ...

PiperOrigin-RevId: 590680938

2023-12-13 12:40:06 -08:00

..

advanced-autodiff.md

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

advanced-compilation.md

Move docs/tutorial to docs/tutorials

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

advanced-debugging.md

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

automatic-differentiation.md

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

automatic-vectorization.md

JAX tutorials: add automatic vectorization

2023-11-21 10:39:02 -08:00

debugging.md

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

external-callbacks.md

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

index.rst

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

installation.md

Upgrade JAX Installation doc

2023-12-04 21:42:13 +00:00

jax-as-accelerated-numpy.md

Move docs/tutorial to docs/tutorials

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

jax-internals.md

Move docs/tutorial to docs/tutorials

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

jit-compilation.md

JAX tutorials: quickstart

2023-11-28 08:28:18 -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

JAX tutorials: quickstart

2023-11-28 08:28:18 -08:00

random-numbers.md

JAX tutorials: pseudorandom numbers

2023-11-21 14:26:06 -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

Upgrade JAX debugging doc

2023-12-13 00:19:41 +00:00

working-with-pytrees.md

Update working-with-pytrees.md

2023-12-13 00:40:42 +00:00