|
__init__.py
|
Layernorm
|
2024-11-03 17:05:33 -08:00 |
|
cross_entropy_loss.py
|
Update cross_entropy_loss.py
|
2024-11-04 19:48:09 -08:00 |
|
fast_lora.py
|
Cohere, Bug fixes (#984)
|
2024-09-03 01:52:32 -07:00 |
|
flex_attention.py
|
Disable Flex Attention
|
2024-10-23 02:58:40 -07: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-04 23:16:01 -08:00 |
