Nino Risteski 6aaf377d3a Update RMS LayerNorm implementation, and list compr. change in chat templates (#1974) ... * Update RMS LayerNorm implementation with optimizations and testing suite * perf: optimize list comprehension in get_ollama_eos_tokens		2025-03-14 02:53:21 -07:00
..
__init__.py	patch_fast_lora	2024-11-20 03:36:41 -08:00
cross_entropy_loss.py	constexpr	2025-03-13 18:23:29 -07:00
fast_lora.py	LoRA	2025-03-01 02:52:20 -08:00
flex_attention.py	Bug fixes (#1516)	2025-01-07 04:23:14 -08:00
geglu.py	Bug fixes (#1891)	2025-03-04 03:55:49 -08:00
layernorm.py	constexpr	2025-03-13 18:23:29 -07:00
rms_layernorm.py	Update RMS LayerNorm implementation, and list compr. change in chat templates (#1974)	2025-03-14 02:53:21 -07:00
rope_embedding.py	Bug fixes (#1891)	2025-03-04 03:55:49 -08:00
swiglu.py	Bug fixes (#1891)	2025-03-04 03:55:49 -08:00
utils.py	bug fix	2025-03-05 03:49:38 -08:00