Feature/optax dion optimizer#9

Open

atsentia wants to merge 7 commits intomicrosoft:mainfrom

atsentia:feature/optax-dion-optimizer

Commits on Aug 4, 2025

Add comprehensive test suite for Dion optimizer. Includes unit tests for core optimizer implementations, numerical stability tests, and cross-implementation comparison tests between Dion and Muon v…
atsentia
committed
Fix test environment and dependency conflicts
atsentia
committed
Added test suite and improved code to enable testing
atsentia
committed
Merge branch 'main' of https://github.com/atsentia/dion
atsentia
committed
Add JAX/Optax implementation of DION optimizer
atsentia
committed
Add JAX/Optax implementation of DION optimizer
atsentia
committed
Add development context and next steps for Optax contribution
atsentia
committed