The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
MIT License
torchview: visualize pytorch models
A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being p...
Code for "V1T: Large-scale mouse V1 response prediction using a Vision Transformer"
PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reprod...
TensorLy: Tensor Learning in Python.
Exploration of Self-Semi-Supervised learning for handling unlabeled data