基于 PyTorch 的医学影像深度学习工具包,提供数据加载、预处理、网络结构和评估等组件,帮研究者快速搭建医疗影像 AI 流程。
用 Triton 重写 LLM 训练中的 RMSNorm、RoPE、交叉熵等关键算子,减少显存占用并提升微调吞吐,兼容 Llama、Mistral、Gemma、Phi 等主流模型。
一套 PyTorch 扩展工具集,提供数据增强、损失函数、训练流水线等常用组件,帮你在做图像分割、检测和分类的原型研发与 Kaggle 比赛时少写重复代码。
…67 和 8828 中,我们优化了自动训练器的保存和加载功能;在 8750 中,我们更新了全局剪切的损失函数,进一步提升了自动并行的效率和准确性。 What's Changed [DCU] high performance LLM train and inference for DCU by @yuguo-Jack in https://