This website requires JavaScript.
Explore
Help
Sign In
gahow
/
xtrain
Watch
1
Star
0
Fork
0
You've already forked xtrain
Code
Issues
Pull Requests
Actions
Packages
Projects
Releases
Wiki
Activity
Files
80602099dc488f45c47d1fd25ac1ace339ed8711
xtrain
/
crates
History
Gahow Wang
80602099dc
test: scale Q/K in flash grad-check for well-conditioned grads
...
Co-Authored-By: Claude Opus 4.8 <
noreply@anthropic.com
>
2026-06-17 23:17:04 +08:00
..
xtrain-autodiff
test: scale Q/K in flash grad-check for well-conditioned grads
2026-06-17 23:17:04 +08:00
xtrain-cuda
cuda: fused flash-attention kernel (fwd + flash-style bwd)
2026-06-17 23:10:25 +08:00
xtrain-distributed
test+bins: flash grad-check, flash==composed, PyTorch parity, --flash flag
2026-06-17 23:10:39 +08:00
xtrain-model
test+bins: flash grad-check, flash==composed, PyTorch parity, --flash flag
2026-06-17 23:10:39 +08:00
xtrain-optim
perf: make xtrain-cuda a regular dep of xtrain-optim (GPU AdamW)
2026-06-15 16:53:52 +08:00
xtrain-tensor
cuda: fused flash-attention kernel (fwd + flash-style bwd)
2026-06-17 23:10:25 +08:00
xtrain-train
test+bins: flash grad-check, flash==composed, PyTorch parity, --flash flag
2026-06-17 23:10:39 +08:00