Now filtering pure_callback's messages via logs rather than at stderr.

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

Closed patrick-kidger closed 2 months ago

patrick-kidger commented 2 months ago

Maybe fixes #844.