MegEngine

Known Issue

训练和推理的GPU显存占用(MiB)各模型有不同程度的增加。

New Features

CUDA

  • 添加大卷积核的 direct conv 实现。
  • 添加 implicit bmm 大卷积核 depthwise conv 的实现。

MegEngine

New Features

CUDA

  • Add implementation of large kernel's direct conv algo.
  • Add implementation of large kernel's depthwise conv by implicit bmm.

项目简介

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

🚀 Github 镜像仓库 🚀

源项目地址

https://github.com/MegEngine/MegEngine

发行版本 37

MegEngine v1.13.1

全部发行版

贡献者 39

全部贡献者

开发语言

  • C++ 79.8 %
  • Cuda 13.8 %
  • Python 4.9 %
  • C 0.9 %
  • CMake 0.5 %