tensor
deep-learning
machine-learning
numpy
gpu
opencl
python
neural-network
autograd
tvm
compiler
javascript
performance
vulkan
metal
rocm
spirv
cupy
chainer
einops
tensorflow
pytorch
keras
jax
nim
multidimensional-arrays
cudnn
gpu-computing
parallel-computing
openmp
ndarray
neural-networks
high-performance-computing
pytorch/pytorch347日前75.4k
Tensors and Dynamic neural networks in Python with strong GPU acceleration
apache/tvm346日前10.9k
Open deep learning compiler stack for cpu, gpu and specialized accelerators
arogozhnikov/einops347日前7.7k
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
mratsim/Arraymancer349日前1.2k
A fast, ergonomic and portable tensor library in Nim with a deep learning focus for CPU, GPU and embedded devices via OpenMP, Cuda and OpenCL backends