cotalks.dev

Optimizing Flax NNX Models with Optax (Part 2)

(link)