Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2025-01-16 00:25:07 UTC
- JSON Representation
https://github.com/nvidia/torch-harmonics
Differentiable signal processing on the sphere for PyTorch
machine-learning pytorch signal-processing sphere
Last synced: 10 Jan 2025
https://github.com/SymbioticLab/FedScale
FedScale is a scalable and extensible open-source federated learning (FL) platform.
benchmark dataset deep-learning deployment distributed federated-learning icml machine-learning mlsys osdi pytorch tensorflow
Last synced: 12 Nov 2024
https://github.com/kyegomez/multimodalmamba
A novel implementation of fusing ViT with Mamba into a fast, agile, and high performance Multi-Modal Model. Powered by Zeta, the simplest AI framework ever.
ai artificial-intelligence attention-mechanism machine-learning mamba ml pytorch ssm torch transformer-architecture transformers zeta
Last synced: 10 Jan 2025
https://github.com/kyegomez/MultiModalMamba
A novel implementation of fusing ViT with Mamba into a fast, agile, and high performance Multi-Modal Model. Powered by Zeta, the simplest AI framework ever.
ai artificial-intelligence attention-mechanism machine-learning mamba ml pytorch ssm torch transformer-architecture transformers zeta
Last synced: 30 Oct 2024
https://github.com/emidan19/deep-tempest
Restoration for TEMPEST images using deep-learning
deep-learning gnu-radio image-restoration pytorch sdr tempest
Last synced: 28 Nov 2024
https://github.com/gordicaleksa/pytorch-neural-style-transfer
Reconstruction of the original paper on neural style transfer (Gatys et al.). I've additionally included reconstruction scripts which allow you to reconstruct only the content or the style of the image - for better understanding of how NST works.
deep-learning deep-learning-tutorial machine-learning neural-style-transfer neural-style-transfer-pytorch neural-style-transfer-tutorial non-photorealistic-rendering python pytorch style-transfer
Last synced: 14 Jan 2025
https://github.com/vlomme/Multi-Tacotron-Voice-Cloning
Phoneme multilingual(Russian-English) voice cloning based on
deep-learning g2p pytorch russian tacotron tensorflow tts voice-cloning wavernn
Last synced: 27 Nov 2024
https://github.com/datawhalechina/d2l-ai-solutions-manual
《动手学深度学习》习题解答,在线阅读地址如下:
Last synced: 12 Jan 2025
https://github.com/operatorai/modelstore
🏬 modelstore is a Python library that allows you to version, export, and save a machine learning model to your filesystem or a cloud storage provider.
data-science keras machine-learning mlops modelstore python-library pytorch s3-storage scikit-learn tensorflow transformer
Last synced: 12 Jan 2025
https://github.com/owen-liuyuxuan/visualdet3d
Official Repo for Ground-aware Monocular 3D Object Detection for Autonomous Driving / YOLOStereo3D: A Step Back to 2D for Efficient Stereo 3D Detection
computer-vision icra monocular-3d-detection object-detection pytorch stereo-3d-detection
Last synced: 13 Jan 2025
https://github.com/harritaylor/torchvggish
Pytorch port of Google Research's VGGish model used for extracting audio features.
audio-embedding audioset pytorch vggish
Last synced: 12 Jan 2025
https://github.com/he-y/soft-filter-pruning
Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks
model-compression pruning pytorch
Last synced: 14 Jan 2025
https://github.com/Auto1111SDK/Auto1111SDK
An SDK/Python library for Automatic 1111 to run state-of-the-art diffusion models
ai ai-art api automatic1111 deep-learning diffusers diffusion image-generation image-to-image img2img python pytorch stable-diffusion stable-diffusion-webui text-to-image torch txt2img unstable upscaling web
Last synced: 12 Oct 2024
https://github.com/JohnGiorgi/DeCLUTR
The corresponding code from our paper "DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations". Do not hesitate to open an issue if you run into any trouble!
allennlp contrastive-learning metric-learning natural-language-processing pytorch representation-learning self-supervised-learning semantic-search semantic-text-similarity sentence-embeddings sentence-similarity transformers
Last synced: 01 Nov 2024
https://github.com/ternaus/retinaface
The remake of the https://github.com/biubug6/Pytorch_Retinaface
deeplearning face-detection object-detection pytorch retinaface
Last synced: 15 Jan 2025
https://github.com/Tramac/Fast-SCNN-pytorch
A PyTorch Implementation of Fast-SCNN: Fast Semantic Segmentation Network
computer-vision deep-learning fast-scnn pytorch semantic-segmentation
Last synced: 16 Nov 2024
https://github.com/microsoft/ptgnn
A PyTorch Graph Neural Network Library
deep-learning geometric-deep-learning gnn graph-neural-networks pytorch
Last synced: 11 Jan 2025
https://github.com/gordicaleksa/pytorch-gans
My implementation of various GAN (generative adversarial networks) architectures like vanilla GAN (Goodfellow et al.), cGAN (Mirza et al.), DCGAN (Radford et al.), etc.
conditional-gan dc-gan deep-learning deep-learning-tutorial gan-tutorial gans generative-adversarial-nets generative-adversarial-networks machine-learning python pytorch vanilla-gan
Last synced: 14 Jan 2025
https://github.com/osai-ai/tensor-stream
A library for real-time video stream decoding to CUDA memory
c-plus-plus cuda python pytorch video video-processing
Last synced: 14 Nov 2024
https://github.com/nupurkmr9/vision-aided-gan
Ensembling Off-the-shelf Models for GAN Training (CVPR 2022 Oral)
computer-graphics computer-vision gans generative-adversarial-network image-generation pretrained-models pytorch
Last synced: 15 Nov 2024
https://github.com/msamogh/nonechucks
Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!
data-cleaning data-pipeline data-preprocessing data-processing machine-learning preprocessing pytorch torch
Last synced: 14 Nov 2024
https://github.com/SmallMunich/nutonomy_pointpillars
Convert pointpillars Pytorch Model To ONNX for TensorRT Inference
nutonomy onnx pointpillars pytorch tensorrt
Last synced: 28 Oct 2024
https://github.com/texttron/tevatron
Tevatron - A flexible toolkit for neural retrieval research and development.
dense-retrieval dpr flax information-retrieval jax pytorch question-answering transformer
Last synced: 11 Nov 2024
https://github.com/zuoxingdong/lagom
lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.
artificial-intelligence cem cmaes ddpg deep-deterministic-policy-gradient deep-learning deep-reinforcement-learning evolution-strategies machine-learning mujoco policy-gradient ppo proximal-policy-optimization python pytorch reinforcement-learning research sac soft-actor-critic td3
Last synced: 14 Nov 2024
https://github.com/gordicaleksa/pytorch-GANs
My implementation of various GAN (generative adversarial networks) architectures like vanilla GAN (Goodfellow et al.), cGAN (Mirza et al.), DCGAN (Radford et al.), etc.
conditional-gan dc-gan deep-learning deep-learning-tutorial gan-tutorial gans generative-adversarial-nets generative-adversarial-networks machine-learning python pytorch vanilla-gan
Last synced: 12 Nov 2024
https://github.com/filapro/oneformer3d
[CVPR2024] OneFormer3D: One Transformer for Unified Point Cloud Segmentation
3d-instance-segmentation 3d-object-detection 3d-panoptic-segmentation 3d-semantic-segmentation mmdetection3d pytorch s3dis scannet scannet200
Last synced: 14 Jan 2025
https://github.com/mariusbrataas/flowpoints_ml
An intuitive approach to creating deep learning models
deep-learning flowchart js machine-learning python pytorch react tensorflow
Last synced: 15 Nov 2024
https://github.com/OutofAi/2D-Gaussian-Splatting
A 2D Gaussian Splatting paper for no obvious reasons. Enjoy!
2d-gaussian-splatting gaussian-kernel gaussian-splatting jupyter-notebook python pytorch pytorch-implementation
Last synced: 04 Nov 2024
https://github.com/shobrook/sequitur
Library of autoencoders for sequential data
autoencoder autoencoder-model lstm pytorch sequence time-series
Last synced: 11 Jan 2025
https://github.com/pointcept/pointtransformerv2
[NeurIPS'22] An official PyTorch implementation of PTv2.
Last synced: 14 Jan 2025
https://github.com/vietnh1009/Yolo-v2-pytorch
YOLO for object detection tasks
cnn computer-vision deep-learning deep-neural-networks deeplearning object-detection pytorch yolo yolov2 you-only-look-once
Last synced: 01 Nov 2024
https://github.com/gordicaleksa/pytorch-deepdream
PyTorch implementation of DeepDream algorithm (Mordvintsev et al.). Additionally I've included playground.py to help you better understand basic concepts behind the algo.
deep-dream-tutorial deep-learning deep-learning-tutorial deepdream machine-learning python pytorch
Last synced: 13 Jan 2025
https://github.com/neuromorphicprocessorproject/snn_toolbox
Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.
brian brian2 caffe deep-learning deep-neural-networks keras lasagne loihi nest pynn pytorch spiking-neural-networks spinnaker tensorflow
Last synced: 15 Jan 2025
https://github.com/taokong/foveabox
FoveaBox: Beyond Anchor-based Object Detector
anchor-free coco detection foveabox object-detection pytorch
Last synced: 11 Jan 2025
https://github.com/cgnorthcutt/benchmarking-keras-pytorch
🔥 Reproducibly benchmarking Keras and PyTorch models
benchmarking cnn cnn-classification cnn-keras deep-learning imagenet keras keras-tutorials pytorch pytorch-tutorial
Last synced: 15 Jan 2025
https://github.com/felixgwu/img_classification_pk_pytorch
Quickly comparing your image classification models with the state-of-the-art models (such as DenseNet, ResNet, ...)
cifar computer-vision deep-learning pytorch
Last synced: 10 Jan 2025
https://github.com/benedekrozemberczki/appnp
A PyTorch implementation of "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019).
appnp attention deep-learning deep-neural-networks deepwalk gcn graph-attention graph-classification graph-convolutional-neural-networks graph-embedding graph-neural-network iclr machine-learning network-embedding node-embedding node2vec pagerank ppnp pytorch research
Last synced: 14 Jan 2025
https://github.com/wuyifan18/DeepLog
Pytorch Implementation of DeepLog.
anomaly-detection deeplearning pytorch sequence-prediction
Last synced: 05 Nov 2024
https://github.com/cezannec/capsule_net_pytorch
Readable implementation of a Capsule Network as described in "Dynamic Routing Between Capsules" [Hinton et. al.]
capsule-network pytorch pytorch-tutorial
Last synced: 12 Nov 2024
https://github.com/mratsim/Amazon-Forest-Computer-Vision
Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks
computer-vision data-augmentation deep-learning kaggle kaggle-competition keras neural-network-example neural-networks pytorch transfer-learning
Last synced: 15 Nov 2024
https://github.com/yifanzhang713/IA-SSD
Not All Points Are Equal: Learning Highly Efficient Point-based Detectors for 3D LiDAR Point Clouds (CVPR 2022, Oral)
3d-object-detection autonomous-driving ia-ssd object-detection point-cloud pytorch
Last synced: 28 Oct 2024
https://github.com/mratsim/amazon-forest-computer-vision
Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks
computer-vision data-augmentation deep-learning kaggle kaggle-competition keras neural-network-example neural-networks pytorch transfer-learning
Last synced: 15 Jan 2025
https://github.com/ibm/aihwkit
IBM Analog Hardware Acceleration Kit
ai analog-devices cuda neural-networks pytorch
Last synced: 13 Jan 2025
https://github.com/pmixer/sasrec.pytorch
PyTorch(1.6+) implementation of https://github.com/kang205/SASRec
pytorch recommender-system sasrec sequential-models
Last synced: 12 Jan 2025
https://github.com/ReshotAI/gaussian-painters
Gaussian Painters using 3D Gaussian Splatting
3d ai art gaussian-splatting machine-learning nerf painting python pytorch
Last synced: 04 Nov 2024
https://github.com/svip-lab/planarreconstruction
[CVPR'19] Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding
computer-vision cvpr2019 deep-learning pytorch
Last synced: 12 Jan 2025
https://github.com/Haiyang-W/DSVT
[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"
3d-object-detection backbone cvpr2023 point-cloud pytorch transformer
Last synced: 28 Oct 2024
https://github.com/kach/gradient-descent-the-ultimate-optimizer
Code for our NeurIPS 2022 paper
deep-learning differentiable-programming machine-learning neurips neurips-2022 pytorch
Last synced: 12 Jan 2025
https://github.com/atcold/pytorch-cortexnet
PyTorch implementation of the CortexNet predictive model
deep-learning predictive-modeling pytorch self-supervised unsupervised-learning video
Last synced: 14 Jan 2025
https://github.com/Atcold/pytorch-CortexNet
PyTorch implementation of the CortexNet predictive model
deep-learning predictive-modeling pytorch self-supervised unsupervised-learning video
Last synced: 27 Nov 2024
https://github.com/stepankonev/waymo-motion-prediction-challenge-2022-multipath-plus-plus
Solution for Waymo Motion Prediction Challenge 2022. Our implementation of MultiPath++
autonomous-driving autonomous-vehicles cvpr2022 motion-prediction multipath pytorch sdc self-driving-car trajectory-prediction waymo-open-dataset workshop-autonomous-driving
Last synced: 28 Oct 2024
https://github.com/jungjee/rawnet
Official repository for RawNet, RawNet2, and RawNet3
extracted-speaker-embeddings pytorch rawnet speaker-embeddings speaker-verification spk-embd voxceleb2
Last synced: 13 Jan 2025
https://github.com/sgrvinod/a-pytorch-tutorial-to-sequence-labeling
Empower Sequence Labeling with Task-Aware Neural Language Model | a PyTorch Tutorial to Sequence Labeling
co-training conditional-random-fields crf entity-extraction entity-recognition language-model nlp pos-tagger pos-tagging pytorch pytorch-tutorial sequence-labeling sequence-tagger
Last synced: 14 Jan 2025
https://github.com/francescosaveriozuppichini/pytorch-how-and-when-to-use-module-sequential-modulelist-and-moduledict
Code for my medium article
artificial-intelligence deep-learning machine-learning neural-network pytorch
Last synced: 14 Jan 2025
https://github.com/itailang/SampleNet
Differentiable Point Cloud Sampling (CVPR 2020 Oral)
cvpr2020 deep-learning geometry-processing neural-network point-cloud pytorch sampling tensorlow
Last synced: 28 Oct 2024
https://github.com/vanderschaarlab/synthcity
A library for generating and evaluating synthetic tabular data for privacy, fairness and data augmentation.
data-augmentation fairness-ml generative-model machine-learning privacy pytorch synthetic-data tabular-data
Last synced: 15 Jan 2025
https://github.com/dptech-corp/Uni-Fold
An open-source platform for developing protein models beyond AlphaFold.
alphafold alphafold-multimer deep-learning protein-structure pytorch
Last synced: 12 Nov 2024
https://github.com/amanchadha/iseebetter
iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press
cnn convolutional-neural-network convolutional-neural-networks gan gans generative-adversarial-network psnr pytorch recurrent-neural-networks residual-networks resnet resnets rnn self-supervised-learning srgan ssim super-resolution unsupervised-learning video-processing video-super-resolution
Last synced: 12 Jan 2025
https://github.com/Owen-Liuyuxuan/visualDet3D
Official Repo for Ground-aware Monocular 3D Object Detection for Autonomous Driving / YOLOStereo3D: A Step Back to 2D for Efficient Stereo 3D Detection
computer-vision icra monocular-3d-detection object-detection pytorch stereo-3d-detection
Last synced: 28 Oct 2024
https://github.com/NeuromorphicProcessorProject/snn_toolbox
Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.
brian brian2 caffe deep-learning deep-neural-networks keras lasagne loihi nest pynn pytorch spiking-neural-networks spinnaker tensorflow
Last synced: 09 Nov 2024
https://github.com/aitorzip/pytorch-srgan
A modern PyTorch implementation of SRGAN
cnn computer-vision convolutional-neural-networks deep-learning gan generative-adversarial-network pytorch pytorch-gan srgan super-resolution
Last synced: 15 Jan 2025
https://github.com/lightning-ai/dl-fundamentals
Deep Learning Fundamentals -- Code material and exercises
ai deep-learning machine-learning python pytorch
Last synced: 12 Jan 2025
https://github.com/ShirAmir/dino-vit-features
Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".
co-segmentation computer-vision deep-learning dino part-segmentation pytorch semantic-correspondence vision-transformers
Last synced: 04 Nov 2024
https://github.com/lucidrains/axial-attention
Implementation of Axial attention - attending to multi-dimensional data efficiently
artificial-intelligence attention-mechanism deep-learning pytorch
Last synced: 11 Jan 2025
https://github.com/kevinmusgrave/pytorch-adapt
Domain adaptation made easy. Fully featured, modular, and customizable.
computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning
Last synced: 11 Jan 2025
https://github.com/prigoyal/pytorch_memonger
Experimental ground for optimizing memory of pytorch models
Last synced: 15 Jan 2025
https://github.com/HikariTJU/LD
Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)
deep-learning detection knowledge-distillation ld object-detection pytorch
Last synced: 05 Nov 2024
https://github.com/alibaba/lightweight-neural-architecture-search
This is a collection of our zero-cost NAS and efficient vision applications.
automl classfication cvpr2023 deepmad iclr2023 icml-2022 maedet nas neurips-2022 object-detection pytorch training-free-nas zennas zero-cost-nas
Last synced: 15 Nov 2024
https://github.com/Jungjee/RawNet
Official repository for RawNet, RawNet2, and RawNet3
extracted-speaker-embeddings pytorch rawnet speaker-embeddings speaker-verification spk-embd voxceleb2
Last synced: 06 Nov 2024
https://github.com/benedekrozemberczki/APPNP
A PyTorch implementation of "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019).
appnp attention deep-learning deep-neural-networks deepwalk gcn graph-attention graph-classification graph-convolutional-neural-networks graph-embedding graph-neural-network iclr machine-learning network-embedding node-embedding node2vec pagerank ppnp pytorch research
Last synced: 08 Nov 2024
https://github.com/KevinMusgrave/pytorch-adapt
Domain adaptation made easy. Fully featured, modular, and customizable.
computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning
Last synced: 05 Nov 2024
https://github.com/hysts/pytorch_mpiigaze
An unofficial PyTorch implementation of MPIIGaze and MPIIFaceGaze
computer-vision gaze gaze-estimation pytorch
Last synced: 12 Jan 2025
https://github.com/minar09/cp-vton-plus
Official implementation for "CP-VTON+: Clothing Shape and Texture Preserving Image-Based Virtual Try-On", CVPRW 2020
2d-virtual-try-on cp-vton-plus emcom emcom-lab image-based official-implementation pytorch pytorch-implementation seoultech virtual-try-on virtual-tryon vton
Last synced: 12 Jan 2025
https://github.com/wenqifan03/GraphRec-WWW19
Graph Neural Networks for Social Recommendation, WWW'19
deep-social-recommendations graphrec pytorch recommender-systems socal-recommendation social-network
Last synced: 28 Nov 2024
https://github.com/uber-research/deeppruner
DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)
iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision
Last synced: 15 Jan 2025
https://github.com/kakaoenterprise/JORLDY
Repository for Open Source Reinforcement Learning Framework JORLDY
framework gym mlagents pytorch reinforcement-learning
Last synced: 28 Nov 2024
https://github.com/quantylab/rltrader
파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 - 퀀트 투자, 알고리즘 트레이딩을 위한 최첨단 해법 입문 (개정판)
deep-learning keras python pytorch reinforcement-learning tensorflow
Last synced: 10 Jan 2025
https://github.com/kefirski/pytorch_RVAE
Recurrent Variational Autoencoder that generates sequential data implemented with pytorch
deep-learning nlp python pytorch vae
Last synced: 02 Nov 2024
https://github.com/ymfa/seq2seq-summarizer
Pointer-generator reinforced seq2seq summarization in PyTorch
abstractive-summarization attention attention-mechanism coverage coverage-mechanism pointer-network pytorch reinforcement-learning seq2seq summarization summarizer
Last synced: 13 Nov 2024
https://github.com/QUVA-Lab/escnn
Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries
Last synced: 13 Nov 2024
https://github.com/quva-lab/escnn
Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries
Last synced: 10 Jan 2025
https://github.com/ymcui/PERT
PERT: Pre-training BERT with Permuted Language Model
bert nlp plm pre-trained-model pytorch tensorflow transformers
Last synced: 16 Nov 2024
https://github.com/ymcui/pert
PERT: Pre-training BERT with Permuted Language Model
bert nlp plm pre-trained-model pytorch tensorflow transformers
Last synced: 14 Jan 2025
https://github.com/utkuozbulak/pytorch-cnn-adversarial-attacks
Pytorch implementation of convolutional neural network adversarial attack techniques
adversarial-example fast-gradient-sign fooling-images gradient-ascent pytorch pytorch-tutorial
Last synced: 15 Jan 2025
https://github.com/ybabakhin/kaggle_salt_bes_phalanx
Winning solution for the Kaggle TGS Salt Identification Challenge.
deep-learning image-segmentation keras pseudo-labeling pytorch semantic-segmentation u-net
Last synced: 15 Jan 2025
https://github.com/fendouai/pytorch1.0-cn
PyTorch 1.0 官方文档 中文版,欢迎关注微信公众号:磐创AI
pytorch pytorch-nlp pytorch-tutorial
Last synced: 15 Jan 2025
https://github.com/RLE-Foundation/RLeXplore
RLeXplore provides stable baselines of exploration methods in reinforcement learning, such as intrinsic curiosity module (ICM), random network distillation (RND) and rewarding impact-driven exploration (RIDE).
baselines efficient-algorithm exploration-strategy gym machine-learning pybullet pytorch reinforcement-learning robotics toolbox
Last synced: 04 Dec 2024
https://github.com/square/pysurvival
Open source package for Survival Analysis modeling
deep-learning machine-learning numpy python pytorch survival-analysis
Last synced: 12 Nov 2024
https://github.com/ibm/zshot
Zero and Few shot named entity & relationships recognition
ai deep-learning few-shot few-shot-learning machine-learning named-entity-recognition natural-language-processing natural-language-understanding ned ner nlp nlp-library pytorch relation-extraction relationship-extraction spacy transformer zero-shot zero-shot-learning
Last synced: 11 Jan 2025
https://github.com/maum-ai/hififace
Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)
face-swapping pytorch pytorch-lightning wandb
Last synced: 14 Jan 2025
https://github.com/dvl-tum/mot_neural_solver
Official PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking" (CVPR 2020 Oral).
graph-neural-networks multi-object-tracking pytorch
Last synced: 28 Oct 2024
https://github.com/beomi/infinitransformer
Unofficial PyTorch/🤗Transformers(Gemma/Llama3) implementation of Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention
gemma huggingface infinitransformer llama llama3 pytorch transformers
Last synced: 12 Jan 2025
https://github.com/official-stockfish/nnue-pytorch
Stockfish NNUE (Chess evaluation) trainer in Pytorch
chess deep-learning pytorch stockfish
Last synced: 11 Jan 2025
https://github.com/uber-research/DeepPruner
DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)
iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision
Last synced: 31 Oct 2024
https://github.com/megvii-research/MOTRv2
[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
end-to-end multi-object-tracking pytorch transformer
Last synced: 28 Oct 2024
https://github.com/francescosaveriozuppichini/pytorch-deep-learning-template
A Pytorch Computer Vision template to quick start your next project! 🚀🚀
computer-vision deep-learning python pytorch
Last synced: 12 Jan 2025
https://github.com/davidmascharka/tbd-nets
PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning"
deep-learning machine-learning neural-networks pytorch visual-question-answering visualization vqa
Last synced: 14 Jan 2025