Compile Graph for Custom Triton Kernels

[vasqu]

Mamba2 relevant issues:

• https://github.com/state-spaces/mamba/issues/355
• https://github.com/state-spaces/mamba/issues/369

Torch relevant issues:

• https://github.com/pytorch/pytorch/issues/128059 (seems to occur with 2.3.0)
• https://github.com/Dao-AILab/flash-attention/issues/596
• https://github.com/pytorch/pytorch/issues/93468#issuecomment-1748774273

Graph compiling seems to fail due to either torch issues and/or triton-specific unhashables.