search

patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache License 2.0
2.12k stars 142 forks source link

Improves docs for `tree_at` and `tree_check`. #877

Closed patrick-kidger closed 1 month ago

patrick-kidger commented 1 month ago

Supersedes #874. Fixes #872.

patrick-kidger commented 1 month ago

Thanks for the typo catch! Fixed, and merged. :)

jeertmans commented 1 month ago

Thanks!