11 Commits

Author SHA1 Message Date
jax authors
fc1e1d4a65 Add freshness metablock to JAX OSS docs.
PiperOrigin-RevId: 645508135
2024-06-21 14:50:49 -07:00
rajasekharporeddy
75da54e9f4 Fix doc typos 2024-03-13 14:48:31 +05:30
Jake Vanderplas
56267f08dd Copybara import of the project:
--
371c5a45ea08c8e92136761149d0016077a58652 by Jake VanderPlas <jakevdp@google.com>:

pytree doc: add discussion of children vs aux_data

COPYBARA_INTEGRATE_REVIEW=https://github.com/google/jax/pull/15007 from jakevdp:pytree-doc 371c5a45ea08c8e92136761149d0016077a58652
PiperOrigin-RevId: 517149897
2023-03-16 10:00:24 -07:00
Jake VanderPlas
47ec553c40 DOC: add alternative for pytree initialization 2023-02-17 06:04:33 -08:00
Jake VanderPlas
26d9837b36 Switch to new-style f-strings 2022-12-01 09:14:16 -08:00
Jake VanderPlas
c5728cb5d8 DOC: add section on input validation in custom pytrees 2022-04-13 13:50:05 -07:00
elliotwaite
7392a57b75 DOC: many small fixes 2021-08-04 16:55:13 -07:00
Jake VanderPlas
89dc691da7 DOC: fix issue in pytrees.md 2021-04-15 09:57:53 -07:00
Jake VanderPlas
011c5a203e DOC: add register_pytree_node_class example 2021-04-08 08:55:18 -07:00
Jake VanderPlas
6eb761576c Fix cross-reference in pytrees.md 2021-04-05 16:09:04 -07:00
Jake VanderPlas
35983ebba6 Fix broken example in pytrees.rst & migrate to myst-md 2021-02-22 10:03:32 -08:00