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 d7cc4628f6 perf(gopt): opt concat for OpenCL 3 days ago
.github ci(mge): update test script 1 month ago
ci ci(lint): add commit lint and format check 1 month ago
cmake docs(readme): update build md 4 days ago
dnn fix(ci): fixes for ci 1 day ago
imperative ci(imperative): add test requires 1 day ago
lite feat(lite): add lite static all in one 3 weeks ago
scripts feat(src): add MGB_CUSTOM_OP macro 2 days ago
sdk feat(mgb): dump static memory info by tensorboard 1 day ago
src perf(gopt): opt concat for OpenCL 1 day ago
test feat(src): add MGB_CUSTOM_OP macro 2 days ago
third_party feat(mge/third_party): update cutlass version 2 weeks ago
toolchains feat(cmake/arm/cuda): support cmake arm with cuda, 6 months ago
tools refactor(mgb/dnn): refactor enum used in serializing 2 days ago
.gitattributes feat(ci/windows): fix windows ci stability 1 month ago
.gitignore chroe(.gitignore): add more ignore rules for git 1 year ago
.gitmodules opt(cpuifno): move cpuinfo from pytorch to MegEngine 2 weeks ago
ACKNOWLEDGMENTS feat(lite): add header licence 1 month ago
CMakeLists.txt feat(src): add MGB_CUSTOM_OP macro 2 days ago
CODE_OF_CONDUCT.md fix typos 2 months ago
CONTRIBUTOR_LICENSE_AGREEMENT.md MegEngine: Initial commit of MegEngine. 1 year ago
LICENSE chore(mge): update copyright years 7 months ago
README.md Update README.md 4 days ago
README_CN.md Update README.md 4 days ago
commitlint.config.js ci(copybara): add config files and fix format script 1 month ago
logo.png MegEngine: Initial commit of MegEngine. 1 year ago

README.md

MegEngine

English | 中文

MegEngine is a fast, scalable and easy-to-use deep learning framework, with auto-differentiation.


Installation

NOTE: MegEngine now supports Python installation on Linux-64bit/Windows-64bit/MacOS(CPU-Only)-10.14+ 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

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