MegEngine 天元 (@MegEngine)
Guthub

MegEngine

Twitter logo Share LinkedIn logo Share Facebook logo Share

Top repositories

1

MegEngine

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

InferLLM

a lightweight LLM model inference framework
C++
670
star
3

MegCC

MegCC是一个运行时超轻量，高效，移植简单的深度学习模型编译器
C++
473
star
4

MegSpot

MegSpot是一款高效、专业、跨平台的图片&视频对比应用
Vue
459
star
5

MegFlow

Efficient ML solution for long-tailed demands.
Rust
402
star
6

Models

采用MegEngine实现的各种主流深度学习模型
Python
303
star
7

RepLKNet

Official MegEngine implementation of RepLKNet
Python
268
star
8

MegPeak

C++
247
star
9

PMRID

ECCV2020 - Practical Deep Raw Image Denoising on Mobile Devices
Python
199
star
10

mperf

mperf是一个面向移动/嵌入式平台的算子性能调优工具箱
C++
169
star
11

NBNet

NBNet: Noise Basis Learning for Image Denoising with Subspace Projection
Python
148
star
12

YOLOX

MegEngine implementation of YOLOX
Python
106
star
13

Hub

基于旷视研究院领先的深度学习算法，提供满足多业务场景的预训练模型
Python
90
star
14

mgeconvert

MegEngine到其他框架的转换器
Python
67
star
15

ICD

This is the official implementation of the paper "Instance-conditional Knowledge Distillation for Object Detection", based on MegEngine and Pytorch.
Python
57
star
16

MegRay

A communication library for deep learning
C++
48
star
17

GyroFlow

The official MegEngine implementation of the ICCV 2021 paper: GyroFlow: Gyroscope-Guided Unsupervised Optical Flow Learning
Python
44
star
18

Docs

MegEngine Documentations
Python
44
star
19

Documentation

MegEngine Official Documentation
Python
39
star
20

Resource

Jupyter Notebook
32
star
21

OMNet

OMNet: Learning Overlapping Mask for Partial-to-Partial Point Cloud Registration, ICCV 2021, MegEngine implementation.
Python
32
star
22

examples

A set of examples around MegEngine
Python
29
star
23

ECCV2022-RIFE

Official MegEngine Implementation of Real-Time Intermediate Flow Estimation for Video Frame Interpolation
Python
29
star
24

FINet

This is the official MegEngine implementation of FINet: Dual Branches Feature Interaction for Partial-to-Partial Point Cloud Registration, AAAI 2022
Python
20
star
25

MegDiffusion

MegEngine implementation of Diffusion Models.
Python
16
star
26

awesome-megengine

Awesome Resources about MegEngine
15
star
27

cutlass-bak

modified cutlass
C++
14
star
28

End-to-end-ASR-Transformer

An end to end ASR Transformer model training repo
Python
13
star
29

swin-transformer

Swin-Transformer implementation in MegEngine. This is a showcase for training on GPU with less memory by leveraging MegEngine DTR technique.
Python
12
star
30

MegCat

A Deep Learning Project about cats.
11
star
31

NeRF

NeRF implementation in MegEngine
Python
9
star
32

Inference-Demo

推理样例
C++
8
star
33

megenginelite-rs

Rust
7
star
34

MegCookbook

以《解析深度学习——卷积神经网络原理与视觉实践》一书内容为脉络，提供MegEngine具体代码实现示例和项目案例
7
star
35

cheat_sheet_for_pytorch_immigrant

一份给从 PyTorch 迁移过来的用户的小抄
6
star
36

MegEngine-DMVFN

Python
6
star
37

MegEngine-SAM

Python
5
star
38

invis

invisible megengine API
Python
4
star
39

midout

Reduce binary size by removing code blocks
C++
4
star
40

MegRL

A MegEngine implementation of 6 RL algorithms
Python
3
star
41

MegEngine-Benchmark

Python
3
star
42

torch2mge

Python
2
star
43

Transfer-Learning-Library

Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization
Python
2
star
44

xopr

Experimental Operator Library for MegEngine
Python
2
star
45

mperf-libpfm4

forked from https://sourceforge.net/p/perfmon2/libpfm4/ci/master/tree/
C
1
star