0
I Use This!
High Activity

Commits : Listings

Analyzed about 4 hours ago. based on code collected about 4 hours ago.
Aug 17, 2024 — Aug 17, 2025
Commit Message Contributor Files Modified Lines Added Lines Removed Code Location Date
Enable single-controller training paradigm with Pathways (#1140) More... 4 months ago
Set `prevent_cse` to False for most fuji meshes (#1122) More... 4 months ago
update main branch More... 4 months ago
Update runners to support A3 Mega, A3 Ultra, and A4 + add FP8 in Fuji (#1104) More... 4 months ago
[Jax API update] Update jax.core.Primitive to jax.extend.core.Primitive (#1106) More... 4 months ago
Bump minimum transformers version to 4.51.3 (#1131) More... 4 months ago
Re-purpose shared_bastion_name() (#1135) More... 4 months ago
The dtype of `paddings` and `mask` is now `jnp.bool`. (#1133) More... 4 months ago
Fix Splash Attention unittest for sliding window attention. (#1132) More... 4 months ago
Quantizable TransformerFeedForwardMoE layer (#1127) More... 4 months ago
add conv3d torch axlear conversion (#1119) More... 4 months ago
Split quota_path into two variables. (#1128) More... 4 months ago
Isolate open api mocks to fix CI. (#1129) More... 4 months ago
Allows child bundlers to create bundle dir. (#1125) More... 4 months ago
Add v5p meshes for Fuji models (#1121) More... 4 months ago
[easy][DeadCode] Remove unused reserved_tpu config (#1124) More... 4 months ago
Remove optimization barrier (#1123) More... 4 months ago
Adds env_id flag. (#1116) More... 4 months ago
Upgrade grain version. (#1117) More... 4 months ago
Ignore pyenv version file (#1109) More... 4 months ago
Adds support for RaggedTensor in unbatch_dataset. (#1113) More... 4 months ago
ConditionalUpdateTransformation for alternate learner (#1108) More... 4 months ago
[SummaryWriter] Add W&B support for AudioSummary (#1111) More... 4 months ago
Set the logmel upper bound to the Nyquist frequency based on the sample rate. (#1110) More... 4 months ago
Add a Job validator interface and optionally use it if implementation provided to Bastion config (#1107) More... 4 months ago
Some fixes for FP8 training (#1099) More... 4 months ago
Significantly improved performance for GPU FP32 FlashAttention (#1098) More... 4 months ago
fix tree_map More... 4 months ago
Include the year, and job uuid in the job history (#1105) More... 4 months ago
Introduces named runner configs. (#1101) More... 4 months ago