You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Megvii Engine Team bccda5c427 fix(mgb/imperative): fix repeat bug in trace mode 1 week ago
.github feat(ci): update image 1 week ago
ci feat(ci): update ci and readme 6 months ago
cmake feat(mgb/trt): update tensorRT toolchain to 8 1 week ago
dnn fix(dnn): fix reduce sum/mean error when b is large 1 day ago
imperative fix(mgb/imperative): fix repeat bug in trace mode 1 day ago
lite fix(lite): fix input invalid bug in lar for fitting mode 1 day ago
scripts feat(mgb/trt): update tensorRT toolchain to 8 1 week ago
sdk chore(license): remove all license in file header 2 months ago
src fix(lite): fix input invalid bug in lar for fitting mode 1 day ago
test Revert "feat(mgb): add cumprod opr" 1 week ago
third_party feat(mge/third_party): add cudnn-frontend 1 month ago
toolchains feat(mgb): upgrade flatbuffer 1 month ago
tools refactor(imperative/ops): extends DnnOprCaller with template 1 week ago
.clang-format style(dnn): add bypass of clang-format for dnn foreach_opr macro 9 months ago
.gdbinit feat(dispatch): add gdb debug utils 6 months ago
.gitattributes fix(lite): fix input invalid bug in lar for fitting mode 1 day ago
.gitignore chroe(.gitignore): add more ignore rules for git 1 year ago
.gitmodules feat(mge/third_party): add cudnn-frontend 1 month ago
ACKNOWLEDGMENTS feat(lite): add header licence 11 months ago
CMakeLists.txt feat(mgb/trt): update tensorRT toolchain to 8 1 week ago
CODE_OF_CONDUCT.md fix typos 1 year ago
CONTRIBUTOR_LICENSE_AGREEMENT.md MegEngine: Initial commit of MegEngine. 2 years ago
LICENSE chore(mge): update copyright years 1 year ago
README.md docs(Readme): add key features description 1 month ago
README_CN.md docs(README_CN): add key features description 1 month ago
commitlint.config.js ci(copybara): add config files and fix format script 1 year ago
logo.png fix(mge): update readme 2 months ago

README.md

MegEngine

Documentation | 中文文档

MegEngine is a fast, scalable and easy-to-use deep learning framework with 3 key features.

  • Unified core for both training and inference
    • You can represent quantization/dynamic shape/image pre-processing and even derivation in one model.
    • After training, just put everything into your model and inference it on any platform at ease. Speed and precision problems won’t bother you anymore due to the same core inside. Check the usage here.
  • Lowest hardware requirements helped by algorithms
    • In training, GPU memory usage could go down to one-third at the cost of only one additional line, which enables the DTR algorithm.
    • Gain the lowest memory usage when inferencing a model by leveraging our unique pushdown memory planner
  • Inference efficiently on all-platform
    • Inference fast and high-precision on x86/Arm/CUDA/RoCM
    • Support Linux/Windows/iOS/Android/TEE...
    • Save more memory and optimize speed by leveraging advanced usage

Installation

NOTE: MegEngine now supports Python installation on Linux-64bit/Windows-64bit/MacOS(CPU-Only)-10.14+/Android 7+(CPU-Only) platforms with Python from 3.5 to 3.8. On Windows 10 you can either install the Linux distribution through Windows Subsystem for Linux (WSL) or install the Windows distribution directly. Many other platforms are supported for inference.

Binaries

To install the pre-built binaries via pip wheels:

python3 -m pip install --upgrade pip
python3 -m pip install megengine -f https://megengine.org.cn/whl/mge.html

Building from Source

How to Contribute

We strive to build an open and friendly community. We aim to power humanity with AI.

How to Contact Us

Resources

License

MegEngine is licensed under the Apache License, Version 2.0

Citation

If you use MegEngine in your publication,please cite it by using the following BibTeX entry.

@Misc{MegEngine,
  institution = {megvii},
  title =  {MegEngine:A fast, scalable and easy-to-use deep learning framework},
  howpublished = {\url{https://github.com/MegEngine/MegEngine}},
  year = {2020}
}

Copyright (c) 2014-2021 Megvii Inc. All rights reserved.