Sergei Lebedev 01194bd2fb Clarified the type of the inputs to callback APIs ...

The callback APIs were migrated to use jax.Arrays for both inputs and outputs
in JAX 0.4.27.

PiperOrigin-RevId: 634473890

2024-05-16 11:29:09 -07:00

..

advanced-autodiff.md

Fix type in advanced-autodiff.md

2024-05-14 11:52:41 -04:00

advanced-compilation.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

advanced-debugging.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

external-callbacks.md

Clarified the type of the inputs to callback APIs

2024-05-16 11:29:09 -07:00

gradient-checkpointing.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

index.rst

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

jax-primitives.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

jaxpr.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

parallelism.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

profiling-and-performance.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00

simple-neural-network.md

DOC: replace old tutorials with new content

2024-05-03 12:20:06 -07:00