Matthias Lopinski

icon company Karlsruhe Institute of Technology icon location Karlsruhe, Germany icon location Bioengineering student with a passion for computer science and programming

Results 1 comments of Matthias Lopinski

Migrate vae example from flax.linen to flax.nnx

Just to add to this, I noticed the VAE example here uses `nnx`: [Link](https://github.com/google/flax/blob/main/examples/nnx_toy_examples/05_vae.py) I think this speaks to a broader problem with the example structure. As the `nnx` API...