Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP
Extract video features from raw videos using multiple GPUs. We support RAFT flow frames as well as S3D, I3D, R(2+1)D, VGGish, CLIP, and TIMM models.
Multi-threaded GUI manager for mass creation of AI-generated art with support for multiple GPUs.
Face recognition system for ID photos
GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web interface. Includes multi-GPUs support.
A PyTorch implementation of the 'FaceNet' paper for training a facial recognition model with Triplet Loss using the glint360k dataset. A pre-trained model using Triplet Loss is available for download.
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
Chains stable-diffusion-webui instances together to facilitate faster image generation.
multi-gpu pre-training in one machine for BERT without horovod (Data Parallelism)
Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed computing
Neutron: A pytorch based implementation of Transformer and its variants.
🎯 Accumulated Gradients for TensorFlow 2
A multi GPU audio2face blendshape AI model trainer for your iPhone ARKit data.
Deep Neural Network Compression based on Student-Teacher Network
Code repository for training a brain tumour U-Net 3D image segmentation model using the 'Task1 Brain Tumour' medical segmentation decathlon challenge dataset.
MelGAN Multi GPU Implementation.
image retrieval with cosine metric learning
Add a description, image, and links to the multi-gpu topic page so that developers can more easily learn about it.
To associate your repository with the multi-gpu topic, visit your repo's landing page and select "manage topics."