Skip to content

Dev -> Main #885

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 212 commits into from
Sep 5, 2025
Merged

Dev -> Main #885

merged 212 commits into from
Sep 5, 2025

Conversation

@priyakasimbeg
Copy link
Contributor

@priyakasimbeg priyakasimbeg commented Aug 26, 2025 •
edited
Loading

  • Jit Migration
  • Versioning changes

rka97 and others added 30 commits February 20, 2025 15:51
@rka97
Use jax.jit for sharding initial steps
ae48ccd 
Apply it to the MNIST workload and the Nesterov optimizer.
@rka97
Use jax.jit for adamw
eb5cac7
@rka97
Pass yapf checks
82977da
@rka97
CIFAR workload sharding
99545d4
@rka97
librispeech_conformer now running
018711a 
Still need to test out (a) output losses, (b) speed, and (c) look into
other librispeech.
@rka97
fix formatting
fbeb5f1
@rka97
shard default
6e4e7b0
@rka97
start imagenet
4a2c02d
@rka97
remove bn sync in imagenet (jit handles it automatically)
47beba1
@rka97
ImageNet-ViT also works
3a18f19
@rka97
Start working on WMT. OOM error
bd0f565
@rka97
post-rebase, still on wmt
3044efb
@rka97
cache sharding fix
e301c49
@priyakasimbeg
Merge branch 'dev' into jit_switch
e5ed97a
@rka97
target_setting_algorithms sharding, compilation caching
4fcf984 
compilation caching really speeds things up when doing repeated runs.
@rka97
Update tests to correct batch size
d147e39
@rka97
yapf and isort checks..
a2b61be
@priyakasimbeg
Merge branch 'jit_switch' of https://github.com/mlcommons/algorithmic…
be11c23 
…-efficiency into jit_switch
@priyakasimbeg
Merge branch 'dev' into jit_switch
e2a3b5f
@priyakasimbeg
switch fastmri from pmap to jit
a80f4ec
@priyakasimbeg
migrate criteo workload
c39ca51
@priyakasimbeg
update utils function used for sharding conformer
06377d9
@priyakasimbeg
update conformer and deepspeech
9cbe7d9
@priyakasimbeg
debugging
c6ecd67
@priyakasimbeg
debuging
f35690d
@priyakasimbeg
reformatting
848b50c
@priyakasimbeg
reformatting
fb62eae
@priyakasimbeg
reformatting
fe3f9f0
@priyakasimbeg
reformatting
004afbd
@priyakasimbeg
reformatting
f1db3d3
@priyakasimbeg priyakasimbeg requested a review from a team as a code owner August 26, 2025 16:13
@github-actions
Copy link

github-actions bot commented Aug 26, 2025 •
edited
Loading

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

priyakasimbeg and others added 17 commits August 26, 2025 09:21
@priyakasimbeg
Merge pull request #884 from Niccolo-Ajroldi/dropout_subs_fix
0643268 
Dropout Fix - Final Polishing: updated submissions, tests, docs.
@fsschneider
Merge remote-tracking branch 'upstream/dev' into prepare_v0.6
e02e401
@fsschneider
Full pass tweaking documentation
6f63317
@fsschneider
Add pmap to jit switch to v0.6
75fdc24
@fsschneider
Fix GitHub math [no ci]
a4e6d3a
@fsschneider
Add link to Changelog for details and links to rule change discussion…
ad26373 
…s [no CI]
@fsschneider
Augment with links to rules change discussion/justifications [no CI]
f7bd572
@priyakasimbeg
Merge pull request #876 from fsschneider/prepare_v0.6
484e66c 
Prepare for v0.6 release
@priyakasimbeg
fix resnet call function signature
74b95fb
@priyakasimbeg
fix criteo1tb
e716ac9
@priyakasimbeg
temporary fix remove dropout from attention layers
d1770c2
@priyakasimbeg
make attention dropout static
01cfbb0
@priyakasimbeg
Merge pull request #886 from mlcommons/dropout_fixes
24d9815 
Dropout fixes
@priyakasimbeg
fix for wmt
9325826
@priyakasimbeg
Merge pull request #887 from mlcommons/dropout_fixes
8642cca 
fix for wmt
@priyakasimbeg
fix for wmt dropout
dfce4ca
@priyakasimbeg
Merge pull request #888 from mlcommons/dropout_fixes
7d8f609 
fix for wmt dropout
@priyakasimbeg priyakasimbeg merged commit 2c90454 into main Sep 5, 2025
31 of 32 checks passed
@github-actions github-actions bot locked and limited conversation to collaborators Sep 5, 2025
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

5 participants

@priyakasimbeg @rka97 @fsschneider @Niccolo-Ajroldi