1.pytorch / pytorchTensors and Dynamic neural networks in Python with strong GPU accelerationneural-networkautogradgpunumpydeep-learningtensor+2 morePythonOther license105K total31K forks+1,543 starsfrom 103,527 · score 7.34
2.apache / tvmOpen Machine Learning Compiler Frameworkcompilertensordeep-learninggpuopenclmetal+7 morePythonAPACHE-2.014K total4K forks+101 starsfrom 13,785 · score 4.62
3.cupy / cupyNumPy & SciPy for GPUcudacudnncublascusolverncclpython+12 morePythonMIT12K total1.2K forks+44 starsfrom 12,346 · score 3.81
4.seung-lab / cloud-volumeRead and write Neuroglancer datasets programmatically.neuroglancervolumetric-datapythoncloudnumpyconnectomics+14 morePythonBSD-3-CLAUSE176 total56 forks+1 starsfrom 175 · score 0.69