Pytrees — JAX documentation - discu.eu

Hacker News

Pytrees https://jax.readthedocs.io/en/latest/pytrees.html 14 comments 22/5/2023

Linking pages

GitHub - NTT123/pax: A stateful pytree library for training neural networks. https://github.com/NTT123/pax 0 comments
GitHub - google/jaxtyping: Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. https://github.com/google/jaxtyping 0 comments

Related searches:

Search whole site: site:jax.readthedocs.io

Search title: Pytrees — JAX documentation

See how to search.

Submit link to: