Repository Index

Tensors and Dynamic neural networks in Python with strong GPU acceleration

• neural-network
• autograd
• gpu
• numpy

Open Machine Learning Compiler Framework

• compiler
• tensor
• deep-learning
• gpu

NumPy & SciPy for GPU

• cuda
• cudnn
• cublas
• cusolver