What's Changed
- feat(tensor): add
tensor.select_dim(dim, index)(#5246) @crutcher - feat(nn): add fold4d / Fold4d module (inverse of unfold4d) (#5173) @Bellman281
- Update to new cubecl environment (#5251) @nathanielsimard
- fix(docs): correct LBFGSConfig::init doc comment (#5256) @Mikyx-1
- feat(tensor): add QR decomposition (#5102) @anagaev
- feat(fusion): custom fusion optimization (#5240) @nathanielsimard
- feat(quant): support UE4M3 scale params end to end on CPU (#5253) @ThierryCantin-Demers
- fix(burn-cubecl): fix cubek double buffering matmul (#5252) @laggui
- Update revs (#5265) @nathanielsimard
- fix(quant): round quantization scales up instead of to nearest (#5261) @laggui
- fix(tensor):
TensorDataconvert to bool stores by truthiness (#5269) @SamuelBelanger - refactor(dataset)!: replace
Vec<PixelDepth>withPixelData(#5271) @Charles23R - fix(tests): skip some qr tests in f16 (#5272) @laggui
- feat(burn-cubecl): max/min_dim_with_indices in a single launch (#5273) @ThierryCantin-Demers
- update cube (#5275) @louisfd
- Fix: wgpu fused reduce broadcast produced wrong values (#5268) @Charles23R
- Fix/fusion read composition (#5282) @nathanielsimard
- docs(book): remove backend generics, update device, learner, module and add optimizer section (#5276) @laggui
- feat(ndarray): support assign indexing updates for floats and ints (#5245) @mattisonchao
- feat(backend): promote mask_select to a backend op with dispatch (#5283) @Yashiru
- fix(fusion): decline to fuse unsupported quant scale params (#5279) @ThierryCantin-Demers
- feat(nn): add AdaptiveAvgPool3d module (#5115) @jaweed3
- fix(cubecl): use real out_channels in conv_transpose2d autotune key (#5289) @SamuelBelanger
- refactor(tests): add flex features to enable simd or rayon, disable allocation tracking (nicer output) (#5285) @laundmo
- perf(flex): precompute interpolation axis taps (#5274) @huahuadeliaoliao
- chore: update cubek rev w/ matmul fixes (#5291) @laggui
- feat(autotune): centralize roofline bounds registration for matmul, reduce, and attention (#5188) @ThierryCantin-Demers @SamuelBelanger
- update cube (#5299) @louisfd
- fix(dispatch): remove incorrect checkpointing debug assertion in grad_replace (#5294) @4ktLuffy
- fix(docs): correct tensor sort doc examples in orderable.rs (#5298) @4ktLuffy
- fix(docs): correct chained reduction doc examples in numeric.rs (#5301) @4ktLuffy
- fix(xtask): apply the webgpu feature on the wgpu CI runner (#5300) @4ktLuffy
- chore(deps): update cubecl and cubek for
BlockLevel::BlockTensor(unimplemented) (#5303) @ThierryCantin-Demers - fix(cubecl): update rev for constant bitcasts fix (#5306) @laggui
- fix(tensor): topk check for
k <= shape[dim](#5307) @laggui - test(burn-tensor): run the doc examples instead of only compiling them (#5302) @4ktLuffy
- perf(flex): improve add_bias and conv_plane_accumulate autovectorization (#5267) @laundmo
- fix(burn-tch): panic broadcasting a zero-sized dimension on the tch backend (#5305) @hdimer
- Change visibility of optimizer (#5314) @nathanielsimard
- update cubecl/cubek (#5315) @ThierryCantin-Demers
- fix(cubecl): add missing device throughput on rocm and cpu (#5312) @ThierryCantin-Demers
- Feat/autodiff quantized ops (#5317) @nathanielsimard
- Keep metric items on their device instead of syncing them to the host (#5319) @nathanielsimard
- feat(module): generalize weight reparameterization abstraction (#5311) @laggui
- fix(cubecl): allow metadata-only reshape of sub-byte quantized tensors (#5321) @ThierryCantin-Demers
- update cube (#5322) @louisfd
- fix(dispatch): keep a WgpuDevice conversion when wgpu features unify (#5323) @ThierryCantin-Demers
- fix(deps): update cubecl and use correct windows crate versions (#5327) @laggui
- fix(extension): dispatch autodiff ops to the correct backend (#5326) @laggui
- fix: correct some copy-pasta mistakes (#5329) @laggui
- feat(module): add record param group in the collector and allow loading unused params (#5336) @nathanielsimard
- fix(quant): dequantize transposed quantized tensor (#5281) @subotac
- fix(tensor)!: standardize NaN propagation in extrema reductions (#5290) @jcwal1516
- fix(train): flush async metrics before checkpoint/early-stop (#5320) @TsaoLun
- chore: bump to version 0.22.0-pre.2 (#5338) @laggui