Align the jax stack, name the CUDA extra for CUDA, and drop Python 3.11 - #10

Merged

mahdi-shafiei merged 3 commits into

mainfrom

chore/align-jax-ecosystem-and-drop-py311

Aug 29, 2026

Commits on Aug 28, 2026

Keep the soft top-k mask inside its documented bound
mahdi-shafiei
committed
f23aa32
Route benchmark training through the shared optimizer boundary
mahdi-shafiei
committed
636097e
Align the jax stack, name the CUDA extra for CUDA, and drop Python 3.11
mahdi-shafiei
committed
cfa17af