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 55042195d4 chore(winograd): add Convolutionv2 param 3 years ago
..
gen_heuristic MegEngine: Initial commit of MegEngine. 4 years ago
Makefile feat(dnn/cuda): add cutlass nchw4 convolution 3 years ago
gen_cond_take_kern_impls.py feat(mge): add bfloat16 support 4 years ago
gen_cuda_batch_conv_bias_kern_impls.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_cuda_conv_bias_kern_impls.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_each_mode.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_kern_impls.py fix(mgb/rocm): remove begin-internal of rocm 3 years ago
gen_elemwise_multi_type_kern_impls.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_multi_type_utils.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_special_kern_impls.py fix(mgb/rocm): remove begin-internal of rocm 3 years ago
gen_elemwise_utils.py feat(dnn): enable bool comparison 3 years ago
gen_flatbuffers_converter.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_flatbuffers_schema.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_param_defs.py refactor(meg/internal): move interal codes back to megbrain 3 years ago
gen_tablegen.py feat(imperative): auto generated opdef header and python binding 3 years ago
opr_param_defs.py chore(winograd): add Convolutionv2 param 3 years ago

天元(MegEngine)是旷视自主研发的开源深度学习框架,于2020年3月正式在 OpenI 启智社区开源,天元能够帮助开发者高效的完成深度学习算法的设计、训练、部署,有效提升AI研发工作效率。

C++ Cuda Python C CMake other