PyTorch | Ecosyste.ms: Awesome

https://github.com/jayleicn/ClipBERT

[CVPR 2021 Best Student Paper Honorable Mention, Oral] Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning on image-text and video-text tasks.

cvpr2021 pytorch video-question-answering video-retrieval vision-and-language vqa

Last synced: 03 Aug 2024

https://github.com/HumanCompatibleAI/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 10 Nov 2024

https://github.com/mhamilton723/STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences

computer-vision deep-learning iclr2022 pytorch semantic-segmentation unsupervised-learning

Last synced: 03 Aug 2024

https://github.com/ibm/ffdl

Fabric for Deep Learning (FfDL, pronounced fiddle) is a Deep Learning Platform offering TensorFlow, Caffe, PyTorch etc. as a Service on Kubernetes

ai artificial-intelligence caffe deep-learning deep-neural-networks deeplearning ibm-research-ai jupyter keras kubernetes-cluster machine-learning ml model python pytorch storage tensorflow watson

Last synced: 12 Nov 2024

https://github.com/humancompatibleai/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 01 Oct 2024

https://github.com/thunlp/openattack

An Open-Source Package for Textual Adversarial Attack.

adversarial-attacks adversarial-example natural-language-processing nlp pytorch

Last synced: 10 Nov 2024

https://github.com/wkentaro/pytorch-for-numpy-users

PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com

numpy pytorch

Last synced: 13 Nov 2024

https://github.com/braindecode/braindecode

Deep learning software to decode EEG, ECG or MEG signals

deep-learning ecog eeg electrocorticography electroencephalogram electroencephalography magnetoencephalography meg neuroimaging neuroscience python pytorch

Last synced: 12 Nov 2024

https://github.com/NeuroDiffGym/neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

artificial-intelligence boundary-value-problem deep-learning differential-equations initial-value-problem mathematical-modelling neural-networks ode odes pde-solver physics-informed-neural-networks pinn pypi pytorch scientific-computing time-series

Last synced: 13 Nov 2024

https://github.com/antao97/dgcnn.pytorch

A PyTorch implementation of Dynamic Graph CNN for Learning on Point Clouds (DGCNN)

classification point-cloud pytorch segmentation

Last synced: 28 Oct 2024

https://github.com/open-mmlab/mmfewshot

OpenMMLab FewShot Learning Toolbox and Benchmark

few-shot-learning openmmlab pytorch

Last synced: 11 Nov 2024

https://github.com/santosh-gupta/speedtorch

Library for faster pinned CPU <-> GPU transfer in Pytorch

cpu-gpu-transfer cpu-pinned-tensors cuda cuda-tensors cuda-variables cupy data-transfer embeddings embeddings-trained gpu gpu-transfer machine-learning natural-language-processing nlp pinned-cpu-tensors pytorch pytorch-tensors pytorch-variables sparse sparse-modeling

Last synced: 12 Nov 2024

https://github.com/apple/ml-aim

This repository provides the code and model checkpoints of the research paper: Scalable Pre-training of Large Autoregressive Image Models

jax large-scale-vision-models mlx pytorch

Last synced: 07 Oct 2024

https://github.com/qqlu/Entity

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

cnn computer-vision condinst deep-learning detectron2 fcos image-segmentation instance-segmentation object-detection panoptic-segmentation pretrained-models pretrained-weights pretraining pytorch segmentation semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/thunlp/OpenAttack

An Open-Source Package for Textual Adversarial Attack.

adversarial-attacks adversarial-example natural-language-processing nlp pytorch

Last synced: 01 Nov 2024

https://github.com/locuslab/qpth

A fast and differentiable QP solver for PyTorch.

deep-learning machine-learning optimization pytorch quadratic-programming

Last synced: 10 Nov 2024

https://github.com/zdevito/aten

ATen: A TENsor library for C++11

pytorch tensor torch7

Last synced: 01 Nov 2024

https://github.com/iceychris/LibreASR

:speech_balloon: An On-Premises, Streaming Speech Recognition System

asr deep-learning esp32-lyrat fastai python pytorch rnn-transducer speech-recognition

Last synced: 05 Nov 2024

https://github.com/breezedeus/cnstd

CnSTD: 基于 PyTorch/MXNet 的中文/英文场景文字检测（Scene Text Detection）、数学公式检测（Mathematical Formula Detection, MFD）、篇章分析（Layout Analysis）的Python3 包

deep-learning math-formula-detection object-detection ocr python pytorch scene-text-detection text-detection

Last synced: 13 Nov 2024

https://github.com/lucidrains/linear-attention-transformer

Transformer based on a variant of attention that is linear complexity in respect to sequence length

artificial-intelligence attention-mechanism deep-learning pytorch transformer

Last synced: 22 Oct 2024

https://github.com/zhangxiaosong18/FreeAnchor

FreeAnchor: Learning to Match Anchors for Visual Object Detection (NeurIPS 2019)

computer-vision freeanchor neurips-2019 object-detection one-stage pytorch

Last synced: 26 Oct 2024

https://github.com/d-li14/mobilenetv2.pytorch

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

cvpr2018 deep-neural-networks imagenet mobilenetv2 pretrained-models pytorch

Last synced: 14 Nov 2024

https://github.com/koursaros-ai/nboost

NBoost is a scalable, search-api-boosting platform for deploying transformer models to improve the relevance of search results on different platforms (i.e. Elasticsearch)

cloud deep-learning docker elasticsearch helm kubernetes machine-learning microservices nboost nlp proxy python pytorch search-api search-engine semantic-search tensorflow

Last synced: 01 Nov 2024

https://github.com/Santosh-Gupta/SpeedTorch

Library for faster pinned CPU <-> GPU transfer in Pytorch

cpu-gpu-transfer cpu-pinned-tensors cuda cuda-tensors cuda-variables cupy data-transfer embeddings embeddings-trained gpu gpu-transfer machine-learning natural-language-processing nlp pinned-cpu-tensors pytorch pytorch-tensors pytorch-variables sparse sparse-modeling

Last synced: 03 Aug 2024

https://github.com/nnsvs/nnsvs

Neural network-based singing voice synthesis library for research

deep-learning dnn kiritan python pytorch singing-synthesis singing-voice singing-voice-synthesis

Last synced: 05 Aug 2024

https://github.com/astra-vision/MonoScene

[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image

2d-to-3d computer-vision cvpr2022 cvpr22 deep-learning kitti-360 mayavi monocular nyu-depth-v2 occupancy-prediction pytorch semantic-kitti semantic-scene-completion semantic-scene-understanding single-image-reconstruction

Last synced: 28 Oct 2024

https://github.com/cv-rits/MonoScene

[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image

2d-to-3d computer-vision cvpr2022 cvpr22 deep-learning kitti-360 mayavi monocular nyu-depth-v2 occupancy-prediction pytorch semantic-kitti semantic-scene-completion semantic-scene-understanding single-image-reconstruction

Last synced: 10 Aug 2024

https://github.com/zjunlp/PromptKG

PromptKG Family: a Gallery of Prompt Learning & KG-related research works, toolkits, and paper-list.

awsome-list demo-tuning dialogue genkgc knowledge-graph knowledge-informed-prompt-learning lambdakg link-prediction natural-language-processing nlp paper paper-list prompt-tuning promptkg pytorch question-answering relation-extraction retrieval-augmented retrievalre retroprompt

Last synced: 03 Aug 2024

https://github.com/Mariewelt/OpenChem

OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research

computational-biology computational-chemistry deep-learning deep-neural-networks drug-design drug-discovery graph-convolutional-networks machine-learning predictive-modeling pytorch qsar

Last synced: 13 Nov 2024

https://github.com/MaximeVandegar/Papers-in-100-Lines-of-Code

Implementation of papers in 100 lines of code.

3d aes artificial-intelligence deep-learning diffusion-models educational gans generative-model implementation-of-research-paper inverse-rendering machine-learning meta-learning nerf neural-radiance-fields papers python pytorch reinforcement-learning research rl

Last synced: 04 Aug 2024

https://github.com/DeepVAC/deepvac

PyTorch Project Specification.

amp coreml ddp deepvac ncnn onnx python pytorch quantization tensorboard tensorrt torchscript

Last synced: 28 Oct 2024

https://github.com/daniil-osokin/lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.

3d-human-pose cmu-panoptic computer-vision deep-learning human-pose-estimation keypoint-estimation lightweight multi-person-pose-estimation openvino pytorch real-time

Last synced: 14 Nov 2024

https://github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.

3d-human-pose cmu-panoptic computer-vision deep-learning human-pose-estimation keypoint-estimation lightweight multi-person-pose-estimation openvino pytorch real-time

Last synced: 10 Nov 2024

https://github.com/Bobo-y/flexible-yolov5

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam，dcn and so on), and tensorrt

backbone cbam dcnv2 gcn hrnet moblienet neck object-detection ptq pytorch qat resnet shufflenet sparsity swin-transformer tensorrt triton-server yolov3 yolov5

Last synced: 09 Nov 2024

https://github.com/mryab/efficient-dl-systems

Efficient Deep Learning Systems course materials (HSE, YSDA)

cuda deep-learning distributed-training efficient-deep-learning machine-learning ml-infrastructure mlops pytorch

Last synced: 13 Nov 2024

https://github.com/ThibaultGROUEIX/AtlasNet

This repository contains the source codes for the paper "AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation ". The network is able to synthesize a mesh (point cloud + connectivity) from a low-resolution point cloud, or from an image.

3d 3d-deep-learning computer-vision cvpr2018 geometry-processing pytorch

Last synced: 28 Oct 2024

https://github.com/deepvac/deepvac

PyTorch Project Specification.

amp coreml ddp deepvac ncnn onnx python pytorch quantization tensorboard tensorrt torchscript

Last synced: 20 Oct 2024

https://github.com/msurtsukov/neural-ode

Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations

jupyter-notebook neural-ode pytorch vae

Last synced: 04 Nov 2024

https://github.com/lorenmt/mtan

The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].

attention-model multi-task-learning pytorch

Last synced: 03 Aug 2024

https://github.com/lezcano/geotorch

Constrained optimization toolkit for PyTorch

constrained-optimization invertible-neural-networks low-rank manifold-optimization orthogonality positive-definite-matrices positive-semi-definite pytorch

Last synced: 29 Oct 2024

https://github.com/TropComplique/mtcnn-pytorch

Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks

convolutional-neural-networks face-detection pytorch

Last synced: 27 Oct 2024

https://github.com/kaituoxu/Conv-TasNet

A PyTorch implementation of Conv-TasNet described in "TasNet: Surpassing Ideal Time-Frequency Masking for Speech Separation" with Permutation Invariant Training (PIT).

audio-separation conv-tasnet permutation-invariant-training pit pytorch source-separation speech-separation tasnet

Last synced: 02 Nov 2024

https://github.com/hollobit/All-About-the-GAN

All About the GANs(Generative Adversarial Networks) - Summarized lists for GAN

adversarial-nets adversarial-networks arxiv cgan classification ct-denoising deep-learning detection gan generative-adversarial-network image-synthesis medical-image-synthesis medical-imaging neural-network paper pytorch reconstruction segmentation tensorflow unsupervised-learning

Last synced: 02 Nov 2024

https://github.com/norse/norse

Deep learning with spiking neural networks (SNNs) in PyTorch.

autograd deep-learning gpu machine-learning neural-network neuromorphic pytorch pytorch-lightning spiking-neural-networks tensor

Last synced: 13 Oct 2024

https://github.com/ZhendongWang6/Uformer

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

deep-learning image-deblurring image-demoireing image-denoising image-deraining image-restoration pytorch

Last synced: 15 Oct 2024

https://github.com/rasbt/llm-workshop-2024

A 4-hour coding workshop to understand how LLMs are implemented and used

large-language-models llm pytorch

Last synced: 22 Oct 2024

https://github.com/jaywalnut310/glow-tts

A Generative Flow for Text-to-Speech via Monotonic Alignment Search

deep-learning pytorch speech-synthesis text-to-speech tts

Last synced: 09 Nov 2024

https://github.com/iMoonLab/HGNN

Hypergraph Neural Networks (AAAI 2019)

complex-data deep-learning hypergraph hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/ExponentialML/Text-To-Video-Finetuning

Finetune ModelScope's Text To Video model using Diffusers 🧨

deep-learning diffusers diffusion-models modelscope pytorch stable-diffusion text-to-video text2video

Last synced: 10 Nov 2024

https://github.com/swz30/MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

attention-mechanism computer-vision eccv2020 image-denoising image-enhancement image-restoration low-level-vision multi-resolution-streams pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/iMoonLab/DeepHypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/TianhongDai/reinforcement-learning-algorithms

This repository contains most of pytorch implementation based classic deep reinforcement learning algorithms, including - DQN, DDQN, Dueling Network, DDPG, SAC, A2C, PPO, TRPO. (More algorithms are still in progress)

a2c actor-critic algorithm atari2600 ddpg deep-learning deep-reinforcement-learning dqn dueling-dqn flappy-bird ppo proximal-policy-optimization pytorch sac soft-actor-critic trpo trust-region-policy-optimization

Last synced: 07 Aug 2024

https://github.com/pytorch/torcharrow

High performance model preprocessing library on PyTorch

preprocessing python pytorch

Last synced: 07 Oct 2024

https://github.com/fastai/fastai2

Temporary home for fastai v2 while it's being developed

data-science deep-learning fastai jupyter machine-learning nbdev python pytorch

Last synced: 07 Aug 2024

https://github.com/Megvii-BaseDetection/cvpods

All-in-one Toolbox for Computer Vision Research.

3d classification computer-vision cvpods detection keypoints pytorch research segmentation self-supervised-learning

Last synced: 28 Oct 2024

https://github.com/wvangansbeke/LaneDetection_End2End

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars

Last synced: 14 Nov 2024

https://github.com/limbo018/DREAMPlace

Deep learning toolkit-enabled VLSI placement

deep-learning gpu-acceleration pytorch vlsi vlsi-physical-design vlsi-placement

Last synced: 03 Aug 2024

https://github.com/chrischoy/fcgf

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor

Last synced: 07 Nov 2024

https://github.com/blackhc/tfpyth

Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).

machine-learning pytorch tensorflow

Last synced: 09 Nov 2024

https://github.com/BlackHC/tfpyth

Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).

machine-learning pytorch tensorflow

Last synced: 28 Oct 2024

https://github.com/wanmeihuali/taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.

3d-reconstruction 3d-rendering computer-graphics computer-vision machine-learning nerf python pytorch real-time-rendering taichi

Last synced: 07 Nov 2024

https://github.com/wvangansbeke/lanedetection_end2end

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars

Last synced: 07 Nov 2024

https://github.com/demiseom/specaugment

A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain

data-augmentation python pytorch specaugment speech speech-recognition tensorflow

Last synced: 11 Oct 2024

https://github.com/DemisEom/SpecAugment

A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain

data-augmentation python pytorch specaugment speech speech-recognition tensorflow

Last synced: 07 Aug 2024

https://github.com/monologg/JointBERT

Pytorch implementation of JointBERT: "BERT for Joint Intent Classification and Slot Filling"

bert intent-classification joint-bert pytorch slot-filling slu transformers

Last synced: 06 Nov 2024

https://github.com/pytorchbearer/torchbearer

torchbearer: A model fitting library for PyTorch

deep-learning differentiable-programming machine-learning model-fitting python3 pytorch

Last synced: 07 Nov 2024

https://github.com/yuliangxiu/mobilepose

Light-weight Single Person Pose Estimator

data-augmentation dataloader deep-learning deeppose dsntnn heatmap lightweight machine-learning mobile-device mobilenetv2 pose-estimation pytorch real-time realtime resnet-18 shufflenet shufflenet-v2 shufflenetv2 squeezenet

Last synced: 13 Nov 2024

https://github.com/mit-han-lab/pvcnn

[NeurIPS 2019, Spotlight] Point-Voxel CNN for Efficient 3D Deep Learning

frustum-pointnet frustum-pointnets kitti point-cloud point-voxel-cnn pointnet pointnet2 pvcnn pytorch s3dis shapenet

Last synced: 28 Oct 2024

https://github.com/qdrant/quaterion

Blazing fast framework for fine-tuning similarity learning models

contrastive-learning cosine-similarity deep-learning knn machine-learning metric-learning nearest-neighbor-search python pytorch pytorch-lightning similarity-learning similarity-search

Last synced: 01 Nov 2024

https://github.com/lucasjinreal/dcnv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

centernet dcnv2 pytorch

Last synced: 13 Nov 2024

https://github.com/seungwonpark/melgan

MelGAN vocoder (compatible with NVIDIA/tacotron2)

gan neural-vocoder pytorch tts

Last synced: 30 Oct 2024

https://github.com/DerrickXuNu/OpenCOOD

[ICRA 2022] An opensource framework for cooperative detection. Official implementation for OPV2V.

autonomous-driving collaborative-perception computer-vision deep-learning multi-agent-perception multi-agent-systems pytorch simulations

Last synced: 28 Oct 2024

https://github.com/perone/euclidesdb

A multi-model machine learning feature embedding database

cpp database deep-learning machine-learning pytorch search

Last synced: 12 Nov 2024

https://github.com/songlab-cal/tape

Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.

benchmark dataset deep-learning language-modeling protein-sequences protein-structure pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/julesbelveze/time-series-autoencoder

PyTorch Dual-Attention LSTM-Autoencoder For Multivariate Time Series

attention-mechanisms autoencoder forecasting lstm-autoencoder multivariate-timeseries pytorch time-series

Last synced: 20 Oct 2024

https://github.com/imoonlab/deephypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 20 Oct 2024

https://github.com/YuliangXiu/MobilePose

Light-weight Single Person Pose Estimator

data-augmentation dataloader deep-learning deeppose dsntnn heatmap lightweight machine-learning mobile-device mobilenetv2 pose-estimation pytorch real-time realtime resnet-18 shufflenet shufflenet-v2 shufflenetv2 squeezenet

Last synced: 07 Aug 2024

https://github.com/JackWoo0831/Yolov7-tracker

Yolo X, v7, v8 and several Multi-Object Tracker(SORT, DeepSORT, ByteTrack, BoT-SORT, etc.) in MOT17 and VisDrone2019 Dataset. It uses a unified style and integrated tracker for easy embedding in your own projects.

pytorch

Last synced: 28 Oct 2024

https://github.com/ymcui/macbert

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

bert language-model macbert nlp pytorch tensorflow transformers

Last synced: 28 Oct 2024

https://github.com/alibaba/Megatron-LLaMA

Best practice for training LLaMA models in Megatron-LM

deepspeed distributed-training llama llm megatron-lm pretraining pytorch

Last synced: 30 Oct 2024

https://github.com/BobLiu20/YOLOv3_PyTorch

Full implementation of YOLOv3 in PyTorch

object-detection pytorch yolo yolov3

Last synced: 09 Nov 2024

https://github.com/patrick-kidger/neuralcde

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch rough-paths time-series

Last synced: 13 Nov 2024

https://github.com/d-x-y/nas-bench-201

NAS-Bench-201 API and Instruction

automl dataset nas nas-bench-201 pytorch

Last synced: 12 Nov 2024

https://github.com/sgrvinod/a-pytorch-tutorial-to-super-resolution

Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network | a PyTorch Tutorial to Super-Resolution

gan gans generative-adversarial-network pytorch pytorch-tutorial srgan super-resolution

Last synced: 14 Nov 2024

https://github.com/DanAnastasyev/DeepNLP-Course

Deep NLP Course

colab-notebook deep-learning keras nlp pytorch

Last synced: 04 Aug 2024

https://github.com/ternaus/robot-surgery-segmentation

Wining solution and its improvement for MICCAI 2017 Robotic Instrument Segmentation Sub-Challenge

computer-vision deep-learning image-segmentation medical-imaging python pytorch robot-assisted-surgery

Last synced: 01 Nov 2024

https://github.com/YU1ut/MixMatch-pytorch

Code for "MixMatch - A Holistic Approach to Semi-Supervised Learning"

deep-learning pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/grok-ai/nn-template

Generic template to bootstrap your PyTorch project.

best-practices best-practises cookiecutter deep-learning dvc github-actions huggingface-datasets hydra mkdocs pre-commit project-structure pytorch pytorch-lightning reproducibility research streamlit template wandb weights-and-biases

Last synced: 13 Oct 2024

https://github.com/sunshineatnoon/Paper-Implementations

Use PyTorch to implement some classic frameworks

computer-vision deep-learning pytorch

Last synced: 07 Aug 2024

https://github.com/SamsungLabs/ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

hrnets interactive-segmentation pretrained-models pytorch segmentation

Last synced: 03 Aug 2024

https://github.com/patrick-kidger/NeuralCDE

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch rough-paths time-series

Last synced: 02 Nov 2024

https://github.com/blacksamorez/tensor_parallel

Automatically split your PyTorch models on multiple GPUs for training & inference

deep-learning machine-learning natural-language-processing nlp python pytorch pytorch-transformers

Last synced: 05 Nov 2024

https://github.com/petarv-/dgi

Deep Graph Infomax (https://arxiv.org/abs/1809.10341)

deep-graph-infomax graph-convolutional-networks neural-networks python pytorch unsupervised-learning unsupervised-node-embedding

Last synced: 09 Nov 2024

https://github.com/morvanzhou/pytorch-a3c

Simple A3C implementation with pytorch + multiprocessing

a3c actor-critic asynchronous-advantage-actor-critic gym multiprocessing neural-network pytorch toy-example

Last synced: 13 Nov 2024

https://github.com/lcswillems/rl-starter-files

RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code

a2c a3c minigrid multi-process ppo preprocessed-observations pytorch reward-shaping

Last synced: 07 Aug 2024

https://github.com/lucasjinreal/DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

centernet dcnv2 pytorch

Last synced: 27 Oct 2024

https://github.com/lucastabelini/LaneATT

Code for the paper entitled "Keep your Eyes on the Lane: Real-time Attention-guided Lane Detection" (CVPR 2021)

computer-vision deep-learning lane-detection pytorch

Last synced: 28 Oct 2024

https://github.com/chrischoy/FCGF

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor

Last synced: 02 Nov 2024

https://github.com/Ha0Tang/AttentionGAN

AttentionGAN for Unpaired Image-to-Image Translation & Multi-Domain Image-to-Image Translation

adversarial-networks attention-model computer-vision cyclegan deep-learning gans geometry ijcnn image-generation image-to-image-translation image-translation multi-domain pytorch unpaired

Last synced: 03 Aug 2024

https://github.com/mir-group/nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

atomistic-simulations computational-chemistry deep-learning drug-discovery force-fields interatomic-potentials machine-learning materials-science molecular-dynamics pytorch

Last synced: 13 Nov 2024