0
I Use This!
High Activity

Commits : Listings

Analyzed about 9 hours ago. based on code collected about 9 hours ago.
Aug 23, 2024 — Aug 23, 2025
Commit Message Contributor Files Modified Lines Added Lines Removed Code Location Date
Introduce `model_analysis.txt` in trainer. (#824) More... 10 months ago
Quantizer does not return one-hot vectors. (#823) More... 10 months ago
StackOverTime's partial frame is treated as a valid frame, similar to convolution padding. (#822) More... 10 months ago
Stack transformer with skip connection (#821) More... 10 months ago
fix: Instantiate logits_modifier config in sample_decode (#819) More... 10 months ago
upgrade to jax 0.4.34 (#817) More... 10 months ago
Support job in-place update (#816) More... 10 months ago
Strict job names (#814) More... 10 months ago
Fix flash attention layer test (#812) More... 10 months ago
Enable autotune ram for axlearn dataset (#813) More... 10 months ago
[Bug fix] Update gpu flash attention after syntax change and fixed unit tests for flash attention (#809) More... 10 months ago
Fix tfds autotune (#811) More... 10 months ago
Fix Regression from https://github.com/apple/axlearn/pull/737 (#810) More... 10 months ago
Adds Conformer on LibriSpeech experiments. (#805) More... 10 months ago
Add copyright to frontend_benchmark.py (#808) More... 10 months ago
add crash feature (#806) More... 10 months ago
Optimizing the sliding window in the frontend. (#807) More... 10 months ago
add health check (#803) More... 10 months ago
Fix AQT Test Suite & Fix module import (#737) More... 10 months ago
Adds support for shared modules in auto-context handling. (#804) More... 10 months ago
Implements DecoderMixin as a DecodingLayer. (#802) More... 10 months ago
Correct CAUSAL padding in conv when stride > 1. (#801) More... 10 months ago
finish todo after jax upgrade (#800) More... 10 months ago
Adds optional autotune optimization to input_tf_data.sample_from_datasets. (#796) More... 10 months ago
Make some convolution functions public. (#799) More... 10 months ago
Skip GPU test for unsupported unequal length case (#798) More... 10 months ago
Introduce @nowrap annotaion. (#797) More... 10 months ago
Add TPU Monitoring for faster hang detection (#786) More... 10 months ago
Update config.yml (#794) More... 10 months ago
Key and value input for MHA prefill and extend (#793) More... 10 months ago