1.pytorch / pytorchTensors and Dynamic neural networks in Python with strong GPU accelerationneural-networkautogradgpunumpydeep-learningtensor+2 morePythonOther license105K total31K forks+3,330 starsfrom 101,740 · score 8.11
2.apache / tvmOpen Machine Learning Compiler Frameworkcompilertensordeep-learninggpuopenclmetal+7 morePythonAPACHE-2.014K total4K forks+275 starsfrom 13,611 · score 5.62
3.cupy / cupyNumPy & SciPy for GPUcudacudnncublascusolverncclpython+12 morePythonMIT12K total1.2K forks+192 starsfrom 12,198 · score 5.26
4.seung-lab / cloud-volumeRead and write Neuroglancer datasets programmatically.neuroglancervolumetric-datapythoncloudnumpyconnectomics+14 morePythonBSD-3-CLAUSE176 total56 forks+1 starsfrom 175 · score 0.69