Optimizing flux2klein - #458

Open

amepas wants to merge 1 commit into

mainfrom

optimizing-flux2klein

Commits on Aug 14, 2026

feat(flux2klein): Onboard FLUX.2-Klein 4B and 9B with Flax NNX, fast weight loading, causal splash attention, and end-to-end optimizations
amepas
committed
3cb4455