HOTFIX Update JAX MNIST example's load_dataset function call#431

Merged

sudhu2k merged 2 commits intodevfrom

sudhu/mnist_jax_hotfix

Jan 27, 2026

Commits on Jan 26, 2026

Update MNIST dataset loading to use 'ylecun/mnist' instead of 'mnist' for both training and testing datasets.
sugovind
committed

Commits on Jan 27, 2026

Update dataset loading in encoder tests to use 'nyu-mll/glue' for both training and validation datasets.
sugovind
committed