Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/demoriarty/torchpq

Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda

cuda nearest-neighbor-search pytorch

Last synced: 11 Jan 2025

https://github.com/openucx/ucc

Unified Collective Communication Library

collectives cuda deep-learning hpc infiniband mpi openshmem pgas pytorch roce sharp

Last synced: 17 Jan 2025

https://github.com/eric-canas/qreader

Robust and Straight-Forward solution for reading difficult and tricky QR codes within images in Python. Powered by YOLOv8

computer-vision easy-to-use image-processing object-detection pip python pytorch pyzbar qr qrcode qrcode-reader qrcode-scanner yolov8

Last synced: 11 Jan 2025

https://github.com/ajbrock/FreezeOut

Accelerate Neural Net Training by Progressively Freezing Layers

deep-learning densenet machine-learning memes neural-networks pytorch vgg16 wide-residual-networks

Last synced: 15 Nov 2024

https://github.com/DeMoriarty/TorchPQ

Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda

cuda nearest-neighbor-search pytorch

Last synced: 02 Nov 2024

https://github.com/frgfm/torch-scan

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)

benchmark deep-learning deep-neural-networks flops flops-counter keras python pytorch pytorch-utils receptive-field summary

Last synced: 12 Jan 2025

https://github.com/tfjgeorge/nngeometry

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch

ek-fac ekfac fim fisher-information-matrix gram jacobian k-fac kfac neural-tangent-kernel ntk pytorch tangent

Last synced: 13 Jan 2025

https://github.com/jiachens/ModelNet40-C

Repo for "Benchmarking Robustness of 3D Point Cloud Recognition against Common Corruptions" https://arxiv.org/abs/2201.12296

benchmark computer-vision corruption-robustness data-augmentation deep-learning ml-safety point-cloud-processing pytorch regularization robustness

Last synced: 28 Oct 2024

https://github.com/li-plus/dsnet

DSNet: A Flexible Detect-to-Summarize Network for Video Summarization

computer-vision detection machine-learning pytorch video-summarization

Last synced: 27 Dec 2024

https://github.com/stas00/ipyexperiments

Automatic GPU+CPU memory profiling, re-use and memory leaks detection using jupyter/ipython experiment containers

gpu ipython jupyter memory-management notebook pytorch ram

Last synced: 12 Jan 2025

https://github.com/foolwood/dcfnet_pytorch

DCFNet: Discriminant Correlation Filters Network for Visual Tracking

cf end-to-end-learning fft pytorch tracking

Last synced: 16 Jan 2025

https://github.com/vita-group/autospeech

[InterSpeech 2020] "AutoSpeech: Neural Architecture Search for Speaker Recognition" by Shaojin Ding*, Tianlong Chen*, Xinyu Gong, Weiwei Zha, Zhangyang Wang

automl autospeech neural-architecture-search pytorch speaker-recognition

Last synced: 17 Jan 2025

https://github.com/endia-org/Endia

Build and train Neural Networks in Mojo

ai arrays compiler jax machine-learning modular mojo numpy pytorch

Last synced: 09 Dec 2024

https://github.com/nachiket92/PGP

Code for "Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversals," CoRL 2021.

autonomous-vehicles nuscenes pytorch trajectory-prediction

Last synced: 28 Oct 2024

https://github.com/ttanzhiqiang/onnx_tensorrt_project

Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxnet to onnx to tensorrt

batch-inference centerface centernet classify darknet libtorch mxnet onnx-tensorrt pytorch retinaface unet yolor yolov4 yolov5 yolox

Last synced: 27 Oct 2024

https://github.com/mmuckley/torchkbnufft

A high-level, easy-to-deploy non-uniform Fast Fourier Transform in PyTorch.

deep-learning mri nufft pytorch reconstruction

Last synced: 01 Oct 2024

https://github.com/opendilab/di-treetensor

Let DI-treetensor help you simplify the structure processing!(树形运算一不小心就逻辑混乱?DI-treetensor快速帮你搞定)

deep-learning numpy python pytorch reinforcement-learning tensor tree-structure

Last synced: 12 Jan 2025

https://github.com/taited/clip-score

Quick scripts to calculate CLIP text-image similarity

batch clip clip-score pytorch

Last synced: 14 Jan 2025

https://github.com/ssube/onnx-web

web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and AMD

ai-art amd diffusion esrgan flask generative-art gfpgan image-generation linux nvidia python pytorch reactjs stable-diffusion super-resolution text2image upscaling web-app windows

Last synced: 12 Jan 2025

https://github.com/jackaduma/vicuna-lora-rlhf-pytorch

A full pipeline to finetune Vicuna LLM with LoRA and RLHF on consumer hardware. Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the Vicuna architecture. Basically ChatGPT but with Vicuna

chatgpt finetune gpt llama llm lora peft ppo pytorch reward-models rlhf vicuna vicuna-7b

Last synced: 16 Jan 2025

https://github.com/Deepest-Project/MelNet

Implementation of "MelNet: A Generative Model for Audio in the Frequency Domain"

generative-model pytorch tts

Last synced: 05 Nov 2024

https://github.com/soobinseo/Tacotron-pytorch

Pytorch implementation of Tacotron

pytorch tacotron text-to-speech tts

Last synced: 14 Nov 2024

https://github.com/dmarnerides/pydlt

PyTorch based Deep Learning Toolbox

computer-vision deep-learning pytorch toolbox

Last synced: 14 Nov 2024

https://github.com/CyberAgentAILab/layout-dm

LayoutDM: Discrete Diffusion Model for Controllable Layout Generation [Inoue+, CVPR2023]

cvpr2023 diffusion-models generative-ai layout pytorch

Last synced: 31 Oct 2024

https://cyberagentailab.github.io/layout-dm/

LayoutDM: Discrete Diffusion Model for Controllable Layout Generation [Inoue+, CVPR2023]

cvpr2023 diffusion-models generative-ai layout pytorch

Last synced: 31 Oct 2024

https://github.com/islanna/dynamicrelu

Implementation of Dynamic ReLU on Pytorch

dynamic pytorch relu

Last synced: 19 Dec 2024

https://github.com/heronsystems/adeptrl

Reinforcement learning framework to accelerate research

actor-critic artificial-intelligence atari pysc2 pytorch reinforcement-learning starcraft2-ai

Last synced: 12 Jan 2025

https://github.com/albumentations-team/autoalbument

AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/

augmentation automated-machine-learning automl computer-vision deep-learning image-augmentation machine-learning pytorch

Last synced: 14 Jan 2025

https://github.com/sea-snell/implicit-language-q-learning

Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"

implicit-q-learning iql language-model nlp offline-rl python pytorch q-learning reinforcement-learning

Last synced: 29 Dec 2024

https://github.com/lettucecfd/lettuce

Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method

cfd lattice-boltzmann machine-learning physics-simulation pytorch torch

Last synced: 23 Nov 2024

https://github.com/Siyu-C/ACAR-Net

[CVPR 2021] Actor-Context-Actor Relation Network for Spatio-temporal Action Localization

acar-net action-detection activitynet activitynet-challenge arxiv ava ava-dataset ava-kinetics model-zoo pytorch spatio-temporal-action-localization torch

Last synced: 03 Nov 2024

https://github.com/ymcui/lert

LERT: A Linguistically-motivated Pre-trained Language Model(语言学信息增强的预训练模型LERT)

bert lert nlp plm pre-train pytorch tensorflow transformer

Last synced: 06 Dec 2024

https://github.com/bobholamovic/cdlab

Yet another repository for developing and benchmarking deep learning-based change detection methods.

change-detection deep-learning pytorch remote-sensing

Last synced: 17 Jan 2025

https://github.com/nvidia/modulus-sym

Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as higher level abstraction for domain experts

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 15 Jan 2025

https://github.com/Bil369/MaskDetect-YOLOv4-PyTorch

基于PyTorch&YOLOv4实现的口罩佩戴检测 :star: 自建口罩数据集分享

maskdetect pytorch yolov4

Last synced: 09 Nov 2024

https://github.com/aws/sagemaker-pytorch-training-toolkit

Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.

aws docker pytorch sagemaker

Last synced: 17 Jan 2025

https://ku-cvlab.github.io/Self-Attention-Guidance/

Official implementation of the paper "Improving Sample Quality of Diffusion Models Using Self-Attention Guidance" (ICCV 2023)

artificial-intelligence diffusion-models generative-models pytorch

Last synced: 31 Oct 2024

https://github.com/jacobgil/pytorch-zssr

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

computer-vision deep-learning pytorch super-resolution

Last synced: 20 Dec 2024

https://github.com/amazon-science/refined

ReFinED is an efficient and accurate entity linking (EL) system.

entity-extraction entity-linking entity-resolution nlp pytorch

Last synced: 15 Jan 2025

https://github.com/p0p4k/pflowtts_pytorch

Unofficial implementation of NVIDIA P-Flow TTS paper

deep-learning flow-matching pflow pytorch text-to-speech tts

Last synced: 28 Nov 2024

https://github.com/hangg7/deformable-kernels

Deforming kernels to adapt towards object deformation. In ICLR 2020.

iclr2020 operator pytorch representation-learning

Last synced: 20 Dec 2024

https://github.com/matthew-z/r-net

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 19 Dec 2024

https://github.com/matthew-z/R-net

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 02 Nov 2024

https://github.com/yanndubs/neural-process-family

Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.

conditional-neural-process deep-learning machine-learning meta-learning neural-processes pytorch stochastic-processes uncertainty-estimation

Last synced: 15 Jan 2025

https://github.com/catqaq/opentextclassification

OpenTextClassification is all you need for text classification! Open text classification for everyone, enjoy your NLP journey! 这可能是目前为止最全面的开源文本分类项目,支持中英双语、多种模型、多种任务。

bert bert-text-classification chinese-text-classification dpcnn ernie gru lstm multi-label-classification naive-bayes pytorch svm text-classification textcnn textrnn torchtext transformers

Last synced: 16 Jan 2025

https://github.com/kyegomez/visionmamba

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

ai machine-learning mamba pytorch recurrent-neural-network ssm

Last synced: 13 Jan 2025

https://github.com/sml2h3/captcha_trainer_pytorch

基于MobileNetV2/EfficientNet-b0/... + LSTM + CTC的不定长图像识别训练pytorch框架

classfication efficientnet mobilenetv2 pytorch

Last synced: 27 Oct 2024

https://github.com/yerfor/syntaspeech

SyntaSpeech: Syntax-aware Generative Adversarial Text-to-Speech; IJCAI 2022; Official code

gan pytorch speech-synthesis tts

Last synced: 19 Dec 2024

https://github.com/jameschapman19/cca_zoo

Canonical Correlation Analysis Zoo: A collection of Regularized, Deep Learning based, Kernel, and Probabilistic methods in a scikit-learn style framework

canonical-correlation-analysis cca cca-zoo dcca deep kernel multiset-cca multiview pls pytorch tensor-cca

Last synced: 18 Jan 2025

https://github.com/learnables/cherry

A PyTorch Library for Reinforcement Learning Research

learning pytorch reinforcement reinforcement-learning rl

Last synced: 17 Jan 2025

https://github.com/shreyas-bk/u2netdemo

Demonstration using Google Colab to show how U-2-NET can be used for Background Removal, Changing Backgrounds, Bounding Box Creation, Salient Feature Highlighting and Salient Object Cropping.

background-removal bounding-boxes deep-learning image-cropping machine-learning python pytorch saliency-map tensorflow u2net

Last synced: 25 Oct 2024

https://github.com/tugstugi/pytorch-speech-commands

Speech commands recognition with PyTorch | Kaggle 10th place solution in TensorFlow Speech Recognition Challenge

cifar10 classification deep-learning densenet dual-path-networks kaggle neural-network pytorch resnet resnext speech-recognition wide-residual-networks

Last synced: 16 Jan 2025

https://github.com/ericlbuehler/xlora

X-LoRA: Mixture of LoRA Experts

llm lora python pytorch

Last synced: 11 Jan 2025

https://github.com/coloquinte/torchsr

Super Resolution datasets and models in Pytorch

datasets edsr pytorch rcan rdn superresolution torchvision vdsr

Last synced: 14 Jan 2025

https://github.com/francescosaveriozuppichini/resnet

Clean, scalable and easy to use ResNet implementation in Pytorch

convolutional-neural-networks deep-learning neural-networks python3 pytorch

Last synced: 19 Dec 2024

https://github.com/andrewatanov/simclr-pytorch

PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results

contrastive-learning deep-learning pytorch pytorch-implementation representation-learning self-supervised-learning

Last synced: 16 Jan 2025

https://github.com/dusty-nv/jetson-voice

ASR/NLP/TTS deep learning inference library for NVIDIA Jetson using PyTorch and TensorRT

deep-learning jetson jetson-nano nlp pytorch speech-recognition tensorrt text-to-speech

Last synced: 16 Jan 2025

https://github.com/ignf/myria3d

Myria3D: Aerial Lidar HD Semantic Segmentation with Deep Learning

deep-learning geodata lidar-point-cloud pointnet pytorch randla-net semantic-segmentation

Last synced: 13 Jan 2025

https://github.com/Karel911/TRACER

TRACER: Extreme Attention Guided Salient Object Tracing Network (AAAI 2022) implementation in PyTorch

aaai-2022 aaai2022 attention attention-mechanism background-removal image-segmentation pytorch pytorch-implementation salient-object-detection

Last synced: 02 Nov 2024

https://github.com/ikergarcia1996/easy-translate

Easy-Translate is a script for translating large text files with a SINGLE COMMAND. Easy-Translate is designed to be as easy as possible for beginners and as seamlesscustomizable and as possible for advanced users.

4-bit 8-bit begginers cpu easy easy-to-use gpu hugginface hugginface-hub huggingface-transformers llm m2m100 machine-translation nllb200 prompt pytorch quantization transformers translation

Last synced: 12 Jan 2025

https://github.com/rlaphoenix/vsgan

PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth

deep-learning editing esrgan gan image-manipulation python pytorch restoration upscaling vapoursynth

Last synced: 12 Jan 2025

https://github.com/minyoungg/pix2latent

Code for: Transforming and Projecting Images into Class-conditional Generative Networks

biggan cma-es gan inverting-images nevergrad pytorch stylegan2

Last synced: 04 Nov 2024

https://github.com/wmcnally/golfdb

GolfDB is a video database for Golf Swing Sequencing, which involves detecting 8 golf swing events in trimmed golf swing videos. This repo demos the baseline model, SwingNet.

action-recognition event-detection golf pytorch temporal video

Last synced: 04 Dec 2024

https://github.com/vt-vl-lab/guided-pix2pix

[ICCV 2019] Guided Image-to-Image Translation with Bi-Directional Feature Transformation

deep-learning gan pix2pix pytorch

Last synced: 16 Jan 2025

https://github.com/hityzy1122/opencv_transforms_torchvision

opencv reimplement for transforms in torchvision

opencv pillow pytorch torchvision

Last synced: 16 Jan 2025

https://minyoungg.github.io/pix2latent/

Code for: Transforming and Projecting Images into Class-conditional Generative Networks

biggan cma-es gan inverting-images nevergrad pytorch stylegan2

Last synced: 11 Nov 2024

https://github.com/dccuchile/CC6204

Material del curso de Deep Learning de la Universidad de Chile

deep-learning pytorch uchile

Last synced: 27 Oct 2024

https://github.com/a-r-j/proteinworkshop

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)

benchmark dataset deep-learning lightning pretraining protein protein-structure pytorch

Last synced: 13 Jan 2025

https://github.com/LeapLabTHU/EfficientTrain

1.5−3.0× lossless training or pre-training speedup. An off-the-shelf, easy-to-implement algorithm for the efficient training of foundation visual backbones.

computer-vision deep-learning efficient-training machine-learning pytorch

Last synced: 05 Nov 2024

https://github.com/rasbt/dora-from-scratch

LoRA and DoRA from Scratch Implementations

finetuning llm pytorch

Last synced: 16 Jan 2025

https://github.com/coqui-ai/trainer

🐸 - A general purpose model trainer, as flexible as it gets

ai data-science deep-learning machine-learning pytorch

Last synced: 12 Jan 2025