pytorch-image-models

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, e

Python32155apache-2.0

yesterday

augmixconvnextdistributed-training

pytorch-grad-cam

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transf

Python10505mit

21 days ago

class-activation-mapscomputer-visiondeep-learning

ros2_pytorch

ROS2 PyTorch template node - running PyTorch C++ models in ROS2

C++39apache-2.0

6 years ago

pytorch-fid

Compute FID scores with PyTorch.

Python3393apache-2.0

4 months ago

deep-learningfidfid-score

flownet2-pytorch

flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with

Python3101other

last year

pytorch-frame

pytorch-frame

Tabular Deep Learning Library for PyTorch

Python537mit

22 days ago

data-framedeep-learningpytorch

segmentation_models.pytorch

segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-

Python9641mit

2 days ago

computer-visiondeeplab-v3-plusdeeplabv3

ros2_pytorch_cuda

ROS2 PyTorch template node - running PyTorch C++ models in ROS2

C++5apache-2.0

2 years ago

pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Python30191mit

last year

deep-learningneural-networkspytorch

tnt

A lightweight library for PyTorch training tools and utilities

Python1663other

23 hours ago

deep-learningmachine-learningneural-network

deep-high-resolution-net.pytorch

deep-high-resolution-net.pytorch

The project is an official implementation of our CVPR2019 paper "Deep High-Resol

Cuda4324mit

2 months ago

coco-keypoints-detectiondeep-high-resolution-netdeep-learning

pytorch-lightning

pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero cod

Python28337apache-2.0

2 days ago

aiartificial-intelligencedata-science

beauty-net

A simple, flexible, and extensible template for PyTorch. It's beautiful.

Python187mit

last year

cnncomputer-visiondeep-learning

lucent

lucent

Lucid library adapted for PyTorch

Python607apache-2.0

5 months ago

DrQA

DrQA

A pytorch implementation of Reading Wikipedia to Answer Open-Domain Questions.

Python401

3 years ago

drqapytorchsquad

lightnlg

A minimalistic codebase for training NLG models from HuggingFace Transformers us

Python3apache-2.0

2 years ago

bartgenerationgpt-2

tch-rs

Rust bindings for the C++ api of PyTorch.

Rust4063apache-2.0

4 months ago

deep-learningmachine-learningneural-network

deep-learning-project-template

Pytorch Lightning code guideline for conferences

Python1240apache-2.0

last year

kaolin

kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

Python4369apache-2.0

3 months ago

3d-deep-learningartificial-intelligencecamera-api

quickvision

quickvision

An Easy To Use PyTorch Computer Vision Library

Python51apache-2.0

last year

computer-visiondeep-learningpytorch

neural-dream

neural-dream

PyTorch implementation of DeepDream algorithm

Python132mit

3 years ago

computer-visiondeep-dreamdeepdream

pytorch_geometric

pytorch_geometric

Graph Neural Network Library for PyTorch

Python21318mit

yesterday

deep-learninggeometric-deep-learninggraph-convolutional-networks

NDScala

N-dimensional / multi-dimensional arrays (tensors) in Scala 3. Think NumPy ndarr

Scala47agpl-3.0

2 years ago

array-programmingdottymachine-learning

CNN_keras

CNN | Keras/PyTorch | CAPTCHA recognition(卷积神经网络、Keras/PyTorch框架、验证码识别)

Python287mit

5 years ago

captchacnn-kerasmachine-learning

pytracking

pytracking

Visual tracking library based on PyTorch.

Python3170gpl-3.0

3 months ago

computer-visionmachine-learningtracking

TorchSeg

TorchSeg

Fast, modular reference implementation and easy training of Semantic Segmentatio

Python1403mit

5 years ago

machine-learning-book

machine-learning-book

Code Repository for Machine Learning with PyTorch and Scikit-Learn

Jupyter Notebook3427mit

3 months ago

deep-learningmachine-learningneural-networks

bindsnet

bindsnet

Simulation of spiking neural networks (SNNs) using PyTorch.

Python1462agpl-3.0

4 months ago

dynamicgpu-computingmachine-learning

skorch

A scikit-learn compatible neural network library that wraps PyTorch

Jupyter Notebook5875bsd-3-clause

3 days ago

hacktoberfesthuggingfacemachine-learning

detecto

detecto

Build fully-functioning computer vision models with PyTorch

Python613mit

4 months ago

computer-visionfaster-rcnnmachine-learning

edsnlp

Modular, fast NLP framework, compatible with Pytorch and spaCy, offering tailore

Python106bsd-3-clause

4 months ago

clinical-data-warehousedeep-learningfast

einops

Flexible and powerful tensor operations for readable and reliable code (for pyto

Python8500mit

26 days ago

chainercupydeep-learning

pytorch2keras

PyTorch to Keras model convertor

Python857mit

2 years ago

converterdartsdeep-learning

poutyne

poutyne

A simplified framework and utilities for PyTorch

Python569lgpl-3.0

4 months ago

data-sciencedeep-learningkeras

deepdetect

deepdetect

Deep Learning API and Server in C++14 support for Caffe, PyTorch,TensorRT, Dlib,

C++2519other

2 days ago

caffedeep-learninggpu

SFA3D

SFA3D

Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The

Python994mit

last year

3d-object-detectionbevmapcenter

torch-points3d

torch-points3d

Pytorch framework for doing deep learning on point clouds.

203other

3 years ago

sinabs

sinabs

A deep learning library for spiking neural networks which is based on PyTorch, f

Python80agpl-3.0

2 days ago

machine-learningpytorchsnn

TResNet

TResNet

Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Arch

Python471apache-2.0

2 years ago

accuracyarchitectureimagenet

ocaml-torch

ocaml-torch

OCaml bindings for PyTorch

OCaml411apache-2.0

23 days ago

artificial-intelligencedeep-learninggpu

pytorch_tabular

pytorch_tabular

A standard framework for modelling Deep Learning Models for tabular data

Python1374mit

4 days ago

deep-learninghacktoberfestmachine-learning

yolo-v5-tflite-model

YOLOv5 - most advanced vision AI model for object detection. Natively implemente

Python17apache-2.0

3 years ago

neural-style-pt

neural-style-pt

PyTorch implementation of neural style transfer algorithm

Python839mit

2 years ago

deep-learningdeep-styleneural-style

pyro

pyro

Deep universal probabilistic programming with Python and PyTorch

Python8548apache-2.0

6 days ago

bayesianbayesian-inferencedeep-learning

yolov3

yolov3

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Python10215agpl-3.0

yesterday

deep-learningmachine-learningobject-detection

yolov5

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Python50720agpl-3.0

yesterday

coremldeep-learningios

SLM-Lab

SLM-Lab

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of t

Python1256mit

2 years ago

a2ca3cbenchmark

deeplake

deeplake

Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangCh

Python8152apache-2.0

yesterday

aicomputer-visioncv

PyTorchCV

PyTorchCV

PyTorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision.

Shell46apache-2.0

6 years ago

transformers

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and

Python134585apache-2.0

24 hours ago

bertdeep-learningflax

determined

determined

Determined is an open-source machine learning platform that simplifies distribut

Go3037apache-2.0

yesterday

data-sciencedeep-learningdistributed-training

graph-2-text

Graph to sequence implemented in Pytorch combining Graph convolutional networks

Python151mit

5 years ago

gradslam

gradslam

gradslam is an open source differentiable dense SLAM library for PyTorch

Python1301mit

last year

3d-reconstructiondeep-learningmachine-learning

GeoDA

GeoDA

This repository contains the official PyTorch implementation of GeoDA algorithm.

Python32

4 years ago

pytorch-image-models

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, e

Python32155apache-2.0

yesterday

augmixconvnextdistributed-training

pytorch-grad-cam

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transf

Python10505mit

21 days ago

class-activation-mapscomputer-visiondeep-learning

ros2_pytorch

ROS2 PyTorch template node - running PyTorch C++ models in ROS2

C++39apache-2.0

6 years ago

pytorch-fid

Compute FID scores with PyTorch.

Python3393apache-2.0

4 months ago

deep-learningfidfid-score

flownet2-pytorch

flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with

Python3101other

last year

pytorch-frame

pytorch-frame

Tabular Deep Learning Library for PyTorch

Python537mit

22 days ago

data-framedeep-learningpytorch

segmentation_models.pytorch

segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-

Python9641mit

2 days ago

computer-visiondeeplab-v3-plusdeeplabv3

ros2_pytorch_cuda

ROS2 PyTorch template node - running PyTorch C++ models in ROS2

C++5apache-2.0

2 years ago

pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Python30191mit

last year

deep-learningneural-networkspytorch

tnt

A lightweight library for PyTorch training tools and utilities

Python1663other

23 hours ago

deep-learningmachine-learningneural-network

deep-high-resolution-net.pytorch

deep-high-resolution-net.pytorch

The project is an official implementation of our CVPR2019 paper "Deep High-Resol

Cuda4324mit

2 months ago

coco-keypoints-detectiondeep-high-resolution-netdeep-learning

pytorch-lightning

pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero cod

Python28337apache-2.0

2 days ago

aiartificial-intelligencedata-science

beauty-net

A simple, flexible, and extensible template for PyTorch. It's beautiful.

Python187mit

last year

cnncomputer-visiondeep-learning

lucent

lucent

Lucid library adapted for PyTorch

Python607apache-2.0

5 months ago

DrQA

DrQA

A pytorch implementation of Reading Wikipedia to Answer Open-Domain Questions.

Python401

3 years ago

drqapytorchsquad

lightnlg

A minimalistic codebase for training NLG models from HuggingFace Transformers us

Python3apache-2.0

2 years ago

bartgenerationgpt-2

tch-rs

Rust bindings for the C++ api of PyTorch.

Rust4063apache-2.0

4 months ago

deep-learningmachine-learningneural-network

deep-learning-project-template

Pytorch Lightning code guideline for conferences

Python1240apache-2.0

last year

kaolin

kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

Python4369apache-2.0

3 months ago

3d-deep-learningartificial-intelligencecamera-api

quickvision

quickvision

An Easy To Use PyTorch Computer Vision Library

Python51apache-2.0

last year

computer-visiondeep-learningpytorch

neural-dream

neural-dream

PyTorch implementation of DeepDream algorithm

Python132mit

3 years ago

computer-visiondeep-dreamdeepdream

pytorch_geometric

pytorch_geometric

Graph Neural Network Library for PyTorch

Python21318mit

yesterday

deep-learninggeometric-deep-learninggraph-convolutional-networks

NDScala

N-dimensional / multi-dimensional arrays (tensors) in Scala 3. Think NumPy ndarr

Scala47agpl-3.0

2 years ago

array-programmingdottymachine-learning

pytracking

pytracking

Visual tracking library based on PyTorch.

Python3170gpl-3.0

3 months ago

computer-visionmachine-learningtracking

TorchSeg

TorchSeg

Fast, modular reference implementation and easy training of Semantic Segmentatio

Python1403mit

5 years ago

machine-learning-book

machine-learning-book

Code Repository for Machine Learning with PyTorch and Scikit-Learn

Jupyter Notebook3427mit

3 months ago

deep-learningmachine-learningneural-networks

bindsnet

bindsnet

Simulation of spiking neural networks (SNNs) using PyTorch.

Python1462agpl-3.0

4 months ago

dynamicgpu-computingmachine-learning

skorch

A scikit-learn compatible neural network library that wraps PyTorch

Jupyter Notebook5875bsd-3-clause

3 days ago

hacktoberfesthuggingfacemachine-learning

detecto

detecto

Build fully-functioning computer vision models with PyTorch

Python613mit

4 months ago

computer-visionfaster-rcnnmachine-learning

edsnlp

Modular, fast NLP framework, compatible with Pytorch and spaCy, offering tailore

Python106bsd-3-clause

4 months ago

clinical-data-warehousedeep-learningfast

einops

Flexible and powerful tensor operations for readable and reliable code (for pyto

Python8500mit

26 days ago

chainercupydeep-learning

pytorch2keras

PyTorch to Keras model convertor

Python857mit

2 years ago

converterdartsdeep-learning

poutyne

poutyne

A simplified framework and utilities for PyTorch

Python569lgpl-3.0

4 months ago

data-sciencedeep-learningkeras

deepdetect

deepdetect

Deep Learning API and Server in C++14 support for Caffe, PyTorch,TensorRT, Dlib,

C++2519other

2 days ago

caffedeep-learninggpu

SFA3D

SFA3D

Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The

Python994mit

last year

3d-object-detectionbevmapcenter

torch-points3d

torch-points3d

Pytorch framework for doing deep learning on point clouds.

203other

3 years ago

sinabs

sinabs

A deep learning library for spiking neural networks which is based on PyTorch, f

Python80agpl-3.0

2 days ago

machine-learningpytorchsnn

TResNet

TResNet

Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Arch

Python471apache-2.0

2 years ago

accuracyarchitectureimagenet

ocaml-torch

ocaml-torch

OCaml bindings for PyTorch

OCaml411apache-2.0

23 days ago

artificial-intelligencedeep-learninggpu

pytorch_tabular

pytorch_tabular

A standard framework for modelling Deep Learning Models for tabular data

Python1374mit

4 days ago

deep-learninghacktoberfestmachine-learning

yolo-v5-tflite-model

YOLOv5 - most advanced vision AI model for object detection. Natively implemente

Python17apache-2.0

3 years ago

neural-style-pt

neural-style-pt

PyTorch implementation of neural style transfer algorithm

Python839mit

2 years ago

deep-learningdeep-styleneural-style

pyro

pyro

Deep universal probabilistic programming with Python and PyTorch

Python8548apache-2.0

6 days ago

bayesianbayesian-inferencedeep-learning

yolov3

yolov3

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Python10215agpl-3.0

yesterday

deep-learningmachine-learningobject-detection

yolov5

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Python50720agpl-3.0

yesterday

coremldeep-learningios

SLM-Lab

SLM-Lab

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of t

Python1256mit

2 years ago

a2ca3cbenchmark

deeplake

deeplake

Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangCh

Python8152apache-2.0

yesterday

aicomputer-visioncv

PyTorchCV

PyTorchCV

PyTorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision.

Shell46apache-2.0

6 years ago

transformers

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and

Python134585apache-2.0

24 hours ago

bertdeep-learningflax

determined

determined

Determined is an open-source machine learning platform that simplifies distribut

Go3037apache-2.0

yesterday

data-sciencedeep-learningdistributed-training

graph-2-text

Graph to sequence implemented in Pytorch combining Graph convolutional networks

Python151mit

5 years ago

gradslam

gradslam

gradslam is an open source differentiable dense SLAM library for PyTorch

Python1301mit

last year

3d-reconstructiondeep-learningmachine-learning

GeoDA

GeoDA

This repository contains the official PyTorch implementation of GeoDA algorithm.

Python32

4 years ago