Ivy Zheng
@ivyzhengx.bsky.social
JAX & Flax @ GoogleDeepmind
Gemini, neural net frameworks & performance
video games, and other random things
she/her, wife of a wife 🌈
Gemini, neural net frameworks & performance
video games, and other random things
she/her, wife of a wife 🌈
For any of the JAX neural network builders out here (or PyTorch users willing to try out new stuff), we've been renovating our Flax API to be simpler and closer to Python semantics.
Check out our website 👇
flax.readthedocs.io/en/latest/nn...
(It even has pretty model rendering with dropdowns)
Check out our website 👇
flax.readthedocs.io/en/latest/nn...
(It even has pretty model rendering with dropdowns)
November 10, 2024 at 7:27 AM
For any of the JAX neural network builders out here (or PyTorch users willing to try out new stuff), we've been renovating our Flax API to be simpler and closer to Python semantics.
Check out our website 👇
flax.readthedocs.io/en/latest/nn...
(It even has pretty model rendering with dropdowns)
Check out our website 👇
flax.readthedocs.io/en/latest/nn...
(It even has pretty model rendering with dropdowns)
Good summer is gone for now
October 27, 2024 at 2:34 AM
Good summer is gone for now