Daniel Han 1e790b03b2 Apply suggestion from @danielhanchen		2026-01-12 21:32:20 -08:00
..
moe	Revert "[pre-commit.ci] auto fixes from pre-commit.com hooks"	2025-12-01 07:24:58 -08:00
__init__.py	Formatting & bug fixes (#3563)	2025-11-07 06:00:22 -08:00
cross_entropy_loss.py	Improve TRL compatibility and GRPO state restore	2026-01-05 07:02:36 +00:00
fast_lora.py	[pre-commit.ci] auto fixes from pre-commit.com hooks	2025-12-30 15:58:41 +00:00
flex_attention.py	Revert "[FIX] Vllm guided decoding params (#3662)"	2025-12-01 05:43:45 -08:00
fp8.py	Fix FBGEMM/CUTLASS errors on SM100 (Blackwell) GPUs	2026-01-08 04:14:53 +00:00
geglu.py	Revert "[FIX] Vllm guided decoding params (#3662)"	2025-12-01 05:43:45 -08:00
layernorm.py	Revert "[FIX] Vllm guided decoding params (#3662)"	2025-12-01 05:43:45 -08:00
rms_layernorm.py	SFT sample packing (#3566)	2025-12-09 17:36:45 -08:00
rope_embedding.py	fastrope fix for zero strided tensors (#3782)	2025-12-28 21:21:48 -08:00
swiglu.py	Apply suggestion from @danielhanchen	2026-01-12 21:32:20 -08:00
utils.py	Revert "[pre-commit.ci] auto fixes from pre-commit.com hooks"	2025-12-01 07:24:58 -08:00