PR #14169 Remove JAX/Flax

Remove JAX/Flax #14169

DN6 merged 7 commits into main from remove-flax

Remove JAX/Flax support

eee60f10

Drop Flax class-name remapping from model_index.json loading

22086d04

Restore Flax training examples

91e57def

github-actions added documentation

github-actions added quantization

github-actions added models

github-actions added tests

github-actions added utils

github-actions added pipelines

github-actions added examples

github-actions added CI

github-actions added schedulers

github-actions added size/L

DN6 requested a review from

yiyixuxu 107 days ago

DN6 requested a review from

sayakpaul 107 days ago

update

73af8f0b

sayakpaul commented on 2026-07-17

sayakpaul approved these changes on 2026-07-21

Merge remote-tracking branch 'origin/main' into remove-flax

592a803f

Repoint scheduler default-config tests off the Flax fixture

dc273b69

Repoint example training tests off the Flax fixture

6984f488

DN6 merged 80bfe77b into main 102 days ago

Reviewers

sayakpaul

yiyixuxu

Assignees

No one assigned

Labels

documentation quantization models tests size/L utils pipelines examples CI schedulers

Milestone

No milestone