Skip to content

Dropout JAX -> dropout_support #864

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
priyakasimbeg merged 40 commits into from
Jun 18, 2025
Merged

Dropout JAX -> dropout_support #864

priyakasimbeg merged 40 commits into from
Jun 18, 2025

Conversation

@priyakasimbeg
Copy link
Contributor

@priyakasimbeg priyakasimbeg commented Apr 17, 2025

No description provided.

@priyakasimbeg priyakasimbeg requested a review from a team as a code owner April 17, 2025 03:21
@github-actions
Copy link

github-actions bot commented Apr 17, 2025 •
edited
Loading

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

priyakasimbeg added 27 commits May 5, 2025 08:28
@priyakasimbeg
remove nan_to_num convertion
aa25e20
@priyakasimbeg
update models with custom dropout layer
85a3578
@priyakasimbeg
add functional dropout for criteo, fastmri, and vit
9354079
@priyakasimbeg
add functional dropout for ogbg
feb9cc5
@priyakasimbeg
modify wmt model for dropout passing
9bba078
@priyakasimbeg
modify wmt model for dropout passing
31f6019
@priyakasimbeg
reformatting and dropout fixes to fastmri and vit
e36d294
@priyakasimbeg
dropout fix for criteo1tb jax
363da8a
@priyakasimbeg
dropout fix for criteo1tb jax
341bf89
@priyakasimbeg
remove aux dropout option from conformer and from init_model_fn signa…
f0c385b 
…ture for fastmri, vit and criteo
@priyakasimbeg
add dropout piping for conformer and deepspeech
7af5c94
@priyakasimbeg
pipe dropout through model_fn
cbd065b
@priyakasimbeg
fix syntax
31babfd
@priyakasimbeg
dropout changes wmt jax
95d67db
@priyakasimbeg
modify dockerfile
2c96b88
@priyakasimbeg
modify docker build script
54786a6
@priyakasimbeg
fsmall fixes
246d68e
@priyakasimbeg
change docker base image to 12.1.1
0c8dd14
@priyakasimbeg
update base image
a78fa66
@priyakasimbeg
small fix
f0019ac
@priyakasimbeg
remove aux_dropout from submission_runner.py
3cb012e
@priyakasimbeg
remove dropout_rate from init_model_fn for all jax workloads
5e192dd
@priyakasimbeg
remove dropout from model initialization call in submission_runner.py
23828cd
@priyakasimbeg
remove dropout check for None and use default instead if not passed
86b8624
@priyakasimbeg
fix to model_fn default dropout value
05bff91
@priyakasimbeg
fixes
f7d99a6
@priyakasimbeg
Merge branch 'dev' into dropout_jax
4f9a4b3
@priyakasimbeg priyakasimbeg merged commit ae8ca68 into dropout_support Jun 18, 2025
28 of 31 checks passed
@github-actions github-actions bot locked and limited conversation to collaborators Jun 18, 2025
@priyakasimbeg priyakasimbeg deleted the dropout_jax branch December 11, 2025 03:15
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.

2 participants

@priyakasimbeg