Liger-Kernel icon indicating copy to clipboard operation
Liger-Kernel copied to clipboard

[feat] Add jamba support

Open yundai424 opened this issue 1 year ago • 0 comments

🚀 The feature, motivation and pitch

model code here -- https://github.com/huggingface/transformers/blob/main/src/transformers/models/jamba/modeling_jamba.py

might be interesting to see how is a triton implementation of mixer forward compared to existing cuda forward too 🤔

Alternatives

No response

Additional context

No response

yundai424 avatar Aug 24 '24 05:08 yundai424