|
__init__.py
|
Layernorm
|
2024-11-03 17:05:33 -08:00 |
|
cross_entropy_loss.py
|
triton_cast
|
2024-11-11 00:17:22 -08:00 |
|
fast_lora.py
|
Cohere, Bug fixes (#984)
|
2024-09-03 01:52:32 -07:00 |
|
flex_attention.py
|
Update flex_attention.py
|
2024-11-06 16:59:15 -08:00 |
|
geglu.py
|
Mistral Nemo 12b (#777)
|
2024-07-18 10:51:10 -07:00 |
|
layernorm.py
|
Layernorm
|
2024-11-03 17:05:33 -08:00 |
|
rms_layernorm.py
|
Disable compiling
|
2024-11-05 00:11:32 -08:00 |
|
rope_embedding.py
|
Disable compiling
|
2024-11-05 00:11:32 -08:00 |
|
swiglu.py
|
Gemma2 (#709)
|
2024-07-02 22:51:01 -07:00 |
|
utils.py
|
Update utils.py
|
2024-11-11 00:37:37 -08:00 |
