Give training its own GPU device - #526

Open

ArthurBrussee wants to merge 8 commits into

mainfrom

Commits on Aug 30, 2026

Give training its own device, and make the kernels backend-generic
ArthurBrussee
committed
8a7d8d0
Make validation readbacks switchable, and say when they are on
ArthurBrussee
committed
85ae85e
Update burn and cubecl
ArthurBrussee
committed
01888a7
Update burn and cubecl, and fix CI
ArthurBrussee
committed
2422aa0
Gate warn_once behind the validation feature
ArthurBrussee
committed
2489c7c
Reach cubecl through burn
ArthurBrussee
committed
65f19e0
Reach cubecl through burn
ArthurBrussee
committed
3ec8b08
Remove redundant device setup comment
ArthurBrussee
committed
a4f87c7