Search Results for "pytorch" - Page 2
Sort By:
Showing 61 open source projects for "pytorch"
View related business solutions-
Save Up to 91% on Cloud Compute With Spot VMsRun batch jobs at 60-91% off with Spot VMs. Long-running workloads get automatic discounts with sustained use.
-
$300 Free Credits to Build on Google CloudStart your next project with $300 in free Google Cloud credit. Spin up VMs, run containers, query petabytes in BigQuery, or build agents with Gemini Enterprise Agent Platform. Once your credits are used, keep building with 20+ always-free tier products including Compute Engine, Cloud Storage, GKE, and Cloud Run functions. No commitment required—just sign up and start building.
-
1
Stanza
Stanford NLP Python library for many human languages
Stanza is a collection of accurate and efficient tools for the linguistic analysis of many human languages. Starting from raw text to syntactic analysis and entity recognition, Stanza brings state-of-the-art NLP models to languages of your choosing. Stanza is a Python natural language analysis package. It contains tools, which can be used in a pipeline, to convert a string containing human language text into lists of sentences and words, to generate base forms of those words, their parts of... -
2
FlashMLA
FlashMLA: Efficient Multi-head Latent Attention Kernels
FlashMLA is a high-performance decoding kernel library designed especially for Multi-Head Latent Attention (MLA) workloads, targeting NVIDIA Hopper GPU architectures. It provides optimized kernels for MLA decoding, including support for variable-length sequences, helping reduce latency and increase throughput in model inference systems using that attention style. The library supports both BF16 and FP16 data types, and includes a paged KV cache implementation with a block size of 64 to... -
3
Python Outlier Detection
A Python toolbox for scalable outlier detection
...Since 2017, PyOD [AZNL19] has been successfully used in numerous academic researches and commercial products [AZHC+21, AZNHL19]. PyOD has multiple neural network-based models, e.g., AutoEncoders, which are implemented in both PyTorch and Tensorflow. PyOD contains multiple models that also exist in scikit-learn. It is possible to train and predict with a large number of detection models in PyOD by leveraging SUOD framework. A benchmark is supplied for select algorithms to provide an overview of the implemented models. In total, 17 benchmark datasets are used for comparison, which can be downloaded at ODDS. -
4
Lightweight' GAN
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Implementation of 'lightweight' GAN proposed in ICLR 2021, in Pytorch. The main contribution of the paper is a skip-layer excitation in the generator, paired with autoencoding self-supervised learning in the discriminator. Quoting the one-line summary "converge on single gpu with few hours' training, on 1024 resolution sub-hundred images". Augmentation is essential for Lightweight GAN to work effectively in a low data setting. -
Build Agents and Models on One PlatformGemini Enterprise Agent Platform is Google Cloud's comprehensive platform for developers to build, scale, govern, and optimize agents and models. Choose from Google's most advanced models and third-party models like Anthropic's Claude Model Family.
-
5
Hasktorch
Tensors and neural networks in Haskell
...Hasktorch is a library for tensors and neural networks in Haskell. It is an independent open source community project which leverages the core C++ libraries shared by PyTorch. -
6
AudioCraft
Audiocraft is a library for audio processing and generation
AudioCraft is a PyTorch library for text-to-audio and text-to-music generation, packaging research models and tooling for training and inference. It includes MusicGen for music generation conditioned on text (and optionally melody) and AudioGen for text-conditioned sound effects and environmental audio. Both models operate over discrete audio tokens produced by a neural codec (EnCodec), which acts like a tokenizer for waveforms and enables efficient sequence modeling. -
7
Transformers4Rec
Transformers4Rec is a flexible and efficient library
Transformers4Rec is an advanced recommendation system library that leverages Transformer models for sequential and session-based recommendations. The library works as a bridge between natural language processing (NLP) and recommender systems (RecSys) by integrating with one of the most popular NLP frameworks, Hugging Face Transformers (HF). Transformers4Rec makes state-of-the-art transformer architectures available for RecSys researchers and industry practitioners. Traditional recommendation... -
8
learn2learn
A PyTorch Library for Meta-learning Research
Learn2Learn is a PyTorch-based library focused on meta-learning and few-shot learning research. It provides reusable components and meta-learning algorithms, making it easier to build, train, and evaluate models that can quickly adapt to new tasks with minimal data. Learn2Learn is widely used in research for tasks such as few-shot classification, reinforcement learning, and optimization. -
9
fastMRI
A large open dataset + tools to speed up MRI scans using ML
...By enabling reconstruction of high-fidelity MR images from significantly fewer measurements, fastMRI aims to make MRI scanning faster, cheaper, and more accessible in clinical settings. The repository provides an open-source PyTorch framework with data loaders, subsampling utilities, reconstruction models, and evaluation metrics, supporting both research reproducibility and practical experimentation. It includes reference implementations for key MRI reconstruction architectures such as U-Net and Variational Networks (VarNet), along with example scripts for model training and evaluation using the PyTorch Lightning framework. ... -
Earn up to 16% annual interest with Nexo.Generate interest, access liquidity without selling, and execute trades seamlessly. All in one platform. Geographic restrictions, eligibility, and terms apply.
-
10
DIG
A library for graph deep learning research
The key difference with current graph deep learning libraries, such as PyTorch Geometric (PyG) and Deep Graph Library (DGL), is that, while PyG and DGL support basic graph deep learning operations, DIG provides a unified testbed for higher level, research-oriented graph deep learning tasks, such as graph generation, self-supervised learning, explainability, 3D graphs, and graph out-of-distribution. If you are working or plan to work on research in graph deep learning, DIG enables you to develop your own methods within our extensible framework, and compare with current baseline methods using common datasets and evaluation metrics without extra efforts. ... -
11
Proteus Model Builder
GUI for training of neural network models for GuitarML Proteus
GUI for easier installation and training of neural network models for guitar amplifiers and pedals, based on the GuitarML Proteus models. These are usable for Proteus, Chowdhury-DSP BYOD and even NeuralPi, on all platforms incl. Linux and RaspberryPi. What is this? GuitarML's work on Proteus, NeuralPi and Proteusboard (hardware) is amazing. https://github.com/GuitarML Yet, it is not easy to wrap your head around if you are not familiar with programming, AI, machine learning, neuronal... -
12
FairScale
PyTorch extensions for high performance and large scale training
...FairScale puts emphasis on correctness and debuggability, offering hook points, logging, and reference examples for common trainer patterns. Although many ideas have since landed in core PyTorch, FairScale remains a valuable reference and a practical toolbox for squeezing more performance out of multi-GPU and multi-node jobs. -
13
albert_zh
Implementation of A Lite Bert For Self-Supervised Learning Language
...It also provides guidance for fine-tuning downstream tasks such as sentence-pair semantic similarity and Chinese classification benchmarks. The repository includes support paths for TensorFlow, PyTorch conversion, Keras loading, TensorFlow 2.0 loading, and TensorFlow Lite deployment for mobile scenarios. Overall, it is useful for Chinese NLP developers who need compact pretrained language models for classification, similarity, and other language understanding tasks. -
14
YOLOV3 Pytorch
This is a source code for yolo3-pytorch
YOLOV3 Pytorch is a PyTorch implementation of the YOLOv3 object detection model built for training, prediction, and evaluation. The repository provides a complete workflow for users who want to train their own object detector with VOC-style data or use pretrained weights. It includes utilities for annotation conversion, anchor generation, image prediction, video prediction, batch prediction, FPS measurement, heatmap output, and mAP evaluation. -
15
YOLOV4 Pytorch
This is a source code for YoloV4-pytorch that can be used to train you
YOLOV4 Pytorch is a PyTorch implementation of the YOLOv4 object detection model for training and running custom detection systems. The repository is structured around practical workflows, including training, prediction, evaluation, anchor generation, model configuration, and dataset annotation utilities. It supports VOC-style datasets and includes scripts for prediction, mAP evaluation, FPS testing, video prediction, batch prediction, and heatmap generation. -
16
PyTorch Transfer-Learning-Library
Transfer Learning Library for Domain Adaptation, Task Adaptation, etc.
TLlib is an open-source and well-documented library for Transfer Learning. It is based on pure PyTorch with high performance and friendly API. Our code is pythonic, and the design is consistent with torchvision. You can easily develop new algorithms or readily apply existing algorithms. We appreciate all contributions. If you are planning to contribute back bug-fixes, please do so without any further discussion. If you plan to contribute new features, utility functions or extensions, please first open an issue and discuss the feature with us. -
17
Facexlib
FaceXlib aims at providing ready-to-use face-related functions
facexlib is a PyTorch-based library providing ready-to-use face-related functions, including detection, alignment, recognition, and more. It integrates state-of-the-art open-source methods for various face processing tasks. -
18
Official YOLOv7
YOLOv7: Trainable bag-of-freebies sets new state-of-the-art
YOLOv7 is the official implementation of the paper “YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors.” It is a PyTorch-based object detection project focused on high speed and strong accuracy for real-time computer vision. The repository provides model definitions, training scripts, testing tools, inference examples, pretrained weights, and deployment-oriented materials. YOLOv7 introduced training-time improvements that raise accuracy without increasing inference cost, which is why the project became important in real-time detection research. ... -
19
Faster-Rcnn
This is a pytorch implementation library of faster-rcnn
Faster-Rcnn is a PyTorch implementation of the Faster R-CNN two-stage object detection model. It is designed for training and evaluating detectors on VOC-format datasets, including VOC07+12 and custom datasets arranged with VOC-style annotations and images. The repository includes scripts for training, prediction, evaluation, annotation generation, and model summary inspection. -
20
Unet
Source code for unet-pytorch, which can train its own model
Unet-pytorch is a PyTorch implementation of U-Net for semantic segmentation workflows. The repository is built around training, prediction, and mIoU evaluation for VOC-style segmentation data and medical-style datasets. It includes scripts for general training, medical dataset training, prediction, annotation handling, model summaries, and evaluation. -
21
DeepLabv3 Plus
Encoder-Decoder with Atrous Separable Convolution
DeepLabv3 Plus is a PyTorch implementation of DeepLabv3+ for semantic segmentation. It implements the encoder-decoder architecture with atrous separable convolution and provides a practical workflow for training, prediction, and mIoU evaluation. The repository supports VOC-style segmentation datasets and includes utilities for annotation generation, JSON dataset conversion, model summary inspection, prediction, and metric calculation. -
22
SVoice (Speech Voice Separation)
We provide a PyTorch implementation of the paper Voice Separation
SVoice is a PyTorch-based implementation of Facebook Research’s study on speaker voice separation as described in the paper “Voice Separation with an Unknown Number of Multiple Speakers.” This project presents a deep learning framework capable of separating mixed audio sequences where several people speak simultaneously, without prior knowledge of how many speakers are present. -
23
YOLOR
implementation of paper - You Only Learn One Representation
YOLOR is the implementation of “You Only Learn One Representation,” a unified network approach for learning explicit and implicit knowledge together. The project focuses on object detection while exploring how a shared representation can support multiple tasks. It builds on the YOLO family and related PyTorch detection work, combining practical detector training with a research idea about unified representations. YOLOR includes model configurations, training code, evaluation scripts, inference tools, and pretrained weights. Its central contribution is the use of implicit knowledge to improve network performance without treating every task as fully separate. ... -
24
TensorNetwork
A library for easy and efficient manipulation of tensor networks
...The library provides automatic path finding and cost estimation, exposing when contractions will explode in memory and suggesting better orders. Because it supports backends such as NumPy, TensorFlow, PyTorch, and JAX, the same model can run on CPUs, GPUs, or TPUs with minimal code changes. Tutorials and visualization helpers make it easier to understand how network topology affects expressive power and computational cost. -
25
PyTorchVideo
A deep learning library for video understanding research
PyTorchVideo is a deep learning library for video understanding, providing modular components and pretrained models for tasks like action recognition, video classification, detection, and self-supervised learning. It is tightly integrated with PyTorch and PyTorch Lightning, offering flexible APIs for building and training spatiotemporal networks. The library includes efficient implementations of state-of-the-art architectures such as SlowFast, X3D, and MViT, optimized for both research prototyping and production inference. It supports video I/O pipelines, data augmentation, distributed training, and mixed precision computation for large-scale experiments. ...
