Refactor/cubecl

[nathanielsimard]

Update to the new cubecl version and fix breaking changes.

Reduces a lot compilation time.

[codecov]

Codecov Report

Attention: Patch coverage is 52.40964% with 158 lines in your changes missing coverage. Please review.

Project coverage is 83.14%. Comparing base (7a19b5f) to head (407753c).
Report is 4 commits behind head on main.

Files with missing lines	Patch %	Lines
...tes/burn-jit/src/kernel/conv/conv2d/gemm/launch.rs	0.00%	57 Missing ⚠️
...it/src/kernel/conv/conv2d/gemm/homogeneous/base.rs	36.84%	24 Missing ⚠️
crates/burn-jit/src/fusion/matmul/args.rs	0.00%	16 Missing ⚠️
crates/burn-jit/src/fusion/matmul/optimization.rs	58.62%	12 Missing ⚠️
crates/burn-jit/src/kernel/comparison.rs	38.46%	8 Missing ⚠️
crates/burn-jit/src/kernel/reduce/naive/kernel.rs	30.00%	7 Missing ⚠️
crates/burn-jit/src/kernel/binary.rs	53.84%	6 Missing ⚠️
crates/burn-jit/src/kernel/pool/max_pool2d.rs	25.00%	6 Missing ⚠️
...s/burn-jit/src/kernel/conv/conv2d/implicit_gemm.rs	69.23%	4 Missing ⚠️
...urn-jit/src/kernel/conv/conv2d/gemm/loader/bias.rs	0.00%	3 Missing ⚠️
... and 13 more

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2654      +/-   ##
==========================================
+ Coverage   81.84%   83.14%   +1.29%     
==========================================
  Files         838      839       +1     
  Lines      107487   107913     +426     
==========================================
+ Hits        87975    89719    +1744     
+ Misses      19512    18194    -1318     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.