A

apple/axlearn

Settings | Report Duplicate

0

I Use This!

High Activity

Commits : Listings

Analyzed about 4 hours ago. based on code collected about 4 hours ago.

Commit Message	Contributor	Files Modified	Lines Added	Lines Removed	Code Location	Date
Aug 17, 2024 — Aug 17, 2025 Showing page 4 of 38 Search / Filter on:
Move KVCache classes to separate files (#1250)	Hanzhi Zhou	More...				2 months ago
Do not materialize full text/image/speech logits (10GB for 16k seq text). (#1236)	Dongseong Hwang	More...				2 months ago
Add more cases to enable sparse core offloading on v6e (#1242)	Hanzhi Zhou	More...				2 months ago
Set default PRNG implementation on pathways (#1241)	Sam Stoelinga	More...				2 months ago
Pathways: jax client on Mac support (#1243)	Sam Stoelinga	More...				2 months ago
Adds host to global specs for AOT. (#1245)	Mark Lee	More...				2 months ago
Fix rfft crash on logmel frontend. (#1244)	Dongseong Hwang	More...				2 months ago
Fix flash mask computation for jax 0.5.3 (#1234)	Hanzhi Zhou	More...				2 months ago
Update pathways image to jax 0.5.3 (#1225)	Karan Goel	More...				2 months ago
Squash commits (#1238)	haely	More...				2 months ago
remove jax_spmd_mode (#1211)	Steboss	More...				2 months ago
[open_api] Avoid using the same client by two active requests (#1240)	altimofeev	More...				2 months ago
Jax 0.5.3 (#1182)	Matthew Hopkins	More...				2 months ago
Update test flash attention test forward (#1226)	Matthew Hopkins	More...				2 months ago
CausalLM: Metrics now return `WeightedScalar` instead of plain scalars. (#1235)	Dongseong Hwang	More...				2 months ago
Fix kv_state propagation (#1232)	qdavid1	More...				3 months ago
Respect the memory kind of sharding in deserialization (#1231)	Lefan Zhang	More...				3 months ago
Raise an error if a job submission fails. (#1233)	Anthony Polloreno	More...				3 months ago
Adds codeowners for axlearn/cloud. (#1230)	Mark Lee	More...				3 months ago
Adds segment_wise to schedule.py (#1228)	Ruoming Pang	More...				3 months ago
Refactor print statemenets (#1224)	haely	More...				3 months ago
[aot] Discard unsupported XLA options in AOT compilation (#1229)	Chang Lan	More...				3 months ago
Add max_len arg to fake_speech_source(). (#1216)	Dongseong Hwang	More...				3 months ago
Differentiate MultiheadInputLinear.Config and MultiheadOutputLinear.Config (#1227)	qdavid1	More...				3 months ago
Support launching CPU jobset (#1223)	muyangyuapple	More...				3 months ago
Dynamically generate FP8 configs for Fuji (#1186)	Samuel Andersen	More...				3 months ago
Update Dockerfile to use ubuntu:22.04 (#1169)	Samuel Andersen	More...				3 months ago
Logmel: use jnp.fft.rfft instead of jnp.fft.fft. (#1221)	Dongseong Hwang	More...				3 months ago
feat: tpu paged attention kernels (#1218)	tt+7	More...				3 months ago
Adds data_source_cls args to array_record_dataset (#1222)	Haoshuo Huang	More...				3 months ago

←
1
2
3
4
5
6
7
8
9
…
37
38
→