Skip to content

Pull requests: tracel-ai/burn

New pull request New
32 Open 3,588 Closed
32 Open 3,588 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

fix(cosine): remove erroneous restart, match PyTorch CosineAnnealingLR behavior
#5191 opened Jul 18, 2026 by DeathSurfing Loading…
1
feat(cubecl): add native GEMM autotune candidate
#5190 opened Jul 18, 2026 by ppodolsky • Draft
3 tasks done
1
Add autotune bounds for attention and reduce; refactor matmul throughput
#5188 opened Jul 17, 2026 by ThierryCantin-Demers Member • Draft
@ThierryCantin-Demers @SamuelBelanger
1
feat: add PixelShuffle and PixelUnshuffle modules
#5184 opened Jul 17, 2026 by Bellman281 Contributor Loading…
2 tasks done
1
feat: add GaussianNLLLoss
#5182 opened Jul 17, 2026 by Bellman281 Contributor Loading…
2 tasks done
1
update cubecl
#5181 opened Jul 17, 2026 by Charles23R Contributor Loading…
1
Fix how param id are collected in burn-store
#5180 opened Jul 17, 2026 by nathanielsimard Member Loading…
1
feat: add TripletMarginLoss
#5175 opened Jul 17, 2026 by Bellman281 Contributor Loading…
2 tasks done
1
feat: add fold4d / Fold4d module (inverse of unfold4d)
#5173 opened Jul 17, 2026 by Bellman281 Contributor Loading…
2 tasks
3
Fix attention broadcast
#5168 opened Jul 16, 2026 by nathanielsimard Member Loading…
1
feat(burn-cubecl): add deterministic feature for reproducible float reductions
#5156 opened Jul 13, 2026 by guoyucode Loading…
4 tasks
7
8 bit AdamW optimizer
#5153 opened Jul 10, 2026 by Redhawk18 Contributor Loading…
2 tasks done
feat(dataset): make image extension matching case-insensitive
#5152 opened Jul 10, 2026 by veezhang Loading…
2 tasks done
1
fix(flex): support GQA/MQA in attention (q_heads % kv_heads == 0)
#5142 opened Jul 8, 2026 by AnayGarodia Loading…
1
1
fix(autodiff): zero-safe cumprod gradient (no division by input)
#5141 opened Jul 8, 2026 by AnayGarodia Loading…
1
3
Improving Flex performance by utilizing SIMD fast paths
#5123 opened Jul 4, 2026 by Andy2887 Contributor • Draft
1
fix(burn-store): prevent CPU exhaustion via pickle memo bomb
#5120 opened Jul 1, 2026 by Infinty-ux Loading…
1
1
Add MSE and MAE metrics for regression tasks
#5119 opened Jul 1, 2026 by Infinty-ux Loading…
2
feat: add AdaptiveAvgPool3d module (#4832)
#5115 opened Jun 28, 2026 by jaweed3 Contributor Loading…
1
perf(fusion): use cubek's dedicated any/all reduce instruction
#5113 opened Jun 27, 2026 by zhan-wei-919 Contributor Loading…
6
Perf/pooling and interpolate optimization
#5112 opened Jun 26, 2026 by SamuelBelanger Contributor Loading…
2 tasks
@ThierryCantin-Demers @SamuelBelanger
3
Support Bitcast Ops for NonInt Kinds
#5105 opened Jun 24, 2026 by skewballfox Contributor • Draft
2 tasks
QR decomposition
#5102 opened Jun 23, 2026 by anagaev Loading…
2 tasks done
2
Use CBOR as a serialization format across the crates
#5100 opened Jun 22, 2026 by nathanielsimard Member • Draft
[wip] fp8 support
#5096 opened Jun 20, 2026 by skewballfox Contributor Loading…
2 tasks
1
ProTip! Exclude everything labeled bug with -label:bug.