Remove TensorFlow dependency and migrate cifar10_resnet to Flax NNX#1508

Merged

copybara-service[bot] merged 3 commits intogoogle-deepmind:mainfrom

selamw1:cifir10-resnet-fix

Apr 22, 2026

Commits on Dec 3, 2025

remove tf dependecy and adjust to use nnx in cifar10_resnet
selamw1
committed

Commits on Jan 22, 2026

Merge branch 'main' into cifir10-resnet-fix
selamw1
authored

Commits on Mar 12, 2026

Update optimizer usage to match Flax 0.11.0 API requirements (model, grads)
selamw1
committed