kernel/ 子模块 — 高性能内核

概述

kernel/ 子模块提供高性能计算内核，使用 Triton 编写，优化关键操作的计算效率和内存使用。

文件列表

文件	说明
__init__.py	模块入口（空文件）
linear_cross_entropy.py	融合的线性层 + 交叉熵损失
kernels.py	Triton 内核实现（entropy 前向/反向）
fp8_kernel.py	FP8 量化内核