[cbHXBY1D]

I've been using Rust's Burn library recently and have avoided writing kernels in CubeCL because it's lack of documentation and my lack of experience. I would love to see some working together here.