Add mHC fused kernels + LigerMHC API + benchmarks#1065

Open

yukiu00 wants to merge 14 commits intolinkedin:mainfrom

yukiu00:feat/mhc-kernel

Commits on Feb 4, 2026

mhc: add kernels, tests, benchmarks, and docs
yukiu00
committed

Commits on Feb 5, 2026

Commits on Feb 8, 2026

Merge branch 'main' into feat/mhc-kernel
yukiu00
authored

Commits on Feb 9, 2026

mhc: address PR review feedback
yukiu00
committed
Refactor MHC tests to simplify imports
yukiu00
committed
mhc: align benchmark with standard framework and fix convergence test skipif
yukiu00
committed
Remove test_mhc_mini_lm.py file and integrate MiniMHCLM class into test_mhc.py for improved organization and maintainability of convergence tests.
yukiu00
committed
mhc: address remaining PR #1065 review feedback
yukiu00
committed
Fix type hinting in mhc.py by replacing `tuple` with `Tuple` for consistency and clarity. Update function signatures in `_post_res_default_meta` and `_post_res_meta` to use `Tuple` from the `typing…
yukiu00
committed
Add LigerMHC to the transformers module exports.
yukiu00
committed
Remove TorchMHCCoeffs class from test_mhc.py
yukiu00
committed
mhc: code quality cleanup across ops, tests, and benchmarks
yukiu00
committed