v0.22.0-pre.2

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): TensorData convert to bool stores by truthiness (#5269) @SamuelBelanger
• refactor(dataset)!: replace Vec<PixelDepth> with PixelData (#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