Kimi K3 is not cheap
alexinch.com20 pointsby ainch22 comments
In flax nnx, what's the idiomatic way to store state on a Module. For example, if I'm handling the carry manually for an nnx.RNN.
How do I restore one of the orbax checkpoints into NNX from this script?
World models, reinforcement learning, evolutionary methods and fast ML code