A lightweight module for Multi-Task Learning in pytorch.
MIT License
Biologically based dynamics for machine learning and deep learning tools for neuroscience.
View model summaries in PyTorch!
A flexible package for multimodal-deep-learning to combine tabular data with text and images usin...
A PyTorch Library for Multi-Task Learning
TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework for All Hardwares, Bac...
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
Deep Learning and Reinforcement Learning Library for Scientists and Engineers
High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...
PyTorch extensions for fast R&D prototyping and Kaggle farming