Migrate/jit/mask

[louisfd]

Mask kernels for #1422

[codecov]

Codecov Report

Attention: Patch coverage is 97.71863% with 6 lines in your changes are missing coverage. Please review.

Project coverage is 85.90%. Comparing base (093cbd3) to head (e34e09f).
Report is 2 commits behind head on main.

Files	Patch %	Lines
crates/burn-jit/src/fusion/tracing/builder.rs	0.00%	5 Missing ⚠️
.../burn-jit/src/codegen/dialect/gpu/vectorization.rs	0.00%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1456      +/-   ##
==========================================
+ Coverage   85.79%   85.90%   +0.10%     
==========================================
  Files         642      644       +2     
  Lines       71271    71678     +407     
==========================================
+ Hits        61150    61573     +423     
+ Misses      10121    10105      -16     

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

[nathanielsimard]

LGTM