google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.63k stars 174 forks source link

Fixed the rendering of equations in jupyter notebooks #836

Closed hmludwig closed 6 months ago

hmludwig commented 6 months ago

Fixed the rendering of math expressions in jupyter notebook by explicitly mentioning the needed extensions (see #829)

Screenshot 2024-02-29 at 14 40 30
fabianp commented 6 months ago

Amazing, thanks! You're on 🔥 @hmludwig !