0
I Use This!
High Activity

Commits : Listings

Analyzed 1 day ago. based on code collected 2 days ago.
Aug 20, 2024 — Aug 20, 2025
Commit Message Contributor Files Modified Lines Added Lines Removed Code Location Date
improve groupnorm (#785) More... 10 months ago
Make RepeatedConformerLayer's repeat field configurable. (#791) More... 10 months ago
Add multislice XLA flag to improve tf summary metric induced latency. (#787) More... 10 months ago
Adds grain input dispatch. (#784) More... 10 months ago
Support Causal Convolution. (#780) More... 10 months ago
cleanup todo (#788) More... 10 months ago
Simplifies host_to_global_device_array. (#781) More... 10 months ago
load llama v3 weights into fuji (#778) More... 10 months ago
Reshard the train state after restoring from builder (#777) More... 10 months ago
Significantly speedup ConfigBase (#774) More... 10 months ago
Checkpointing with grain. (#768) More... 10 months ago
Fix inconsistent paddings in conv layer. (#776) More... 10 months ago
Use fewer bytes for the NumpyMask (#775) More... 10 months ago
Add implicit-dirs to default gcsfuse settings. (#770) More... 10 months ago
Configure output-uploader as a sidecar container (#767) More... 10 months ago
Adds XLA SDC check configuration to the trainer. (#765) More... 10 months ago
audio pillar starts to use einops (#764) More... 10 months ago
Upgrade axlearn to jax 0.4.33 (#759) More... 10 months ago
Adds grain lm eval inputs. (#760) More... 10 months ago
Sparse Sliding Window Attention (#758) More... 10 months ago
Expose NODE_IP to container env (#763) More... 10 months ago
Adds support for unequal lengths for query vs. key-value to attention logic (#761) More... 10 months ago
Preventing datasets instantiation from pretrained configs (#639) More... 10 months ago
add host mount (#755) More... 10 months ago
Speed up Axlearn CI (#688) More... 10 months ago
Enable XLA Output in GKE Job (#752) More... 10 months ago
Fix infinite dataset. (#754) More... 10 months ago
Erase KV before updating during extend_step (#753) More... 10 months ago
Adds axlearn gcp logs command. (#745) More... 10 months ago
Add new env variable to disable axlearn wrap. (#748) More... 10 months ago