A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

Python75mit

7 months ago

adversarial-attacksdistribution-shiftgraph-convolutional-networks

Generate embeddings from large-scale graph-structured data.

Python3313other

5 months ago

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

Python8983mit

2 months ago

class-activation-mapscomputer-visiondeep-learning

PyTorch Geometric Signed Directed is a signed/directed graph neural network exte

Python104mit

5 months ago

deep-learningdirected-networksgnn

Automatic architecture search and hyperparameter optimization for PyTorch

Python2181apache-2.0

6 months ago

automldeep-learningpytorch

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientN

Python28713apache-2.0

2 months ago

augmixconvnextdistributed-training

Compute FID scores with PyTorch.

Python2949apache-2.0

12 months ago

deep-learningfidfid-score

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

Python3055other

11 months ago

Basic Utilities for PyTorch Natural Language Processing (NLP)

Python2203bsd-3-clause

9 months ago

data-loaderdatasetdeep-learning

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Python75975other

2 months ago

autograddeep-learninggpu

Segmentation models with pretrained backbones. PyTorch.

Python8530mit

2 months ago

deeplab-v3-plusdeeplabv3fpn

PyTorch Tutorial for Deep Learning Researchers

Python28459mit

8 months ago

deep-learningneural-networkspytorch

High-level library to help with training and evaluating neural networks in PyTor

Python4424bsd-3-clause

2 months ago

closemberdeep-learninghacktoberfest

Models, data loaders and abstractions for language processing, powered by PyTorc

Python3419bsd-3-clause

2 months ago

data-loaderdatasetdeep-learning

A lightweight library for PyTorch training tools and utilities

Python1589other

3 months ago

deep-learningmachine-learningneural-network

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code ch

Python26446apache-2.0

last month

aiartificial-intelligencedata-science

Data manipulation and transformation for audio signal processing, powered by PyT

Python2341bsd-2-clause

2 months ago

audioaudio-processingio

Hyperparameter Optimization for TensorFlow, Keras and PyTorch

Python1598mit

5 months ago

artificial-intelligencedeep-learninghyperparameter-optimization

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine

Python2344mit

6 months ago

deep-learninggcngnn

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

Python186mit

5 months ago

cnncomputer-visiondeep-learning

A highly efficient implementation of Gaussian Processes in PyTorch

Python3301mit

4 months ago

gaussian-processesgpu-accelerationpytorch

Lucid library adapted for PyTorch

Python568apache-2.0

4 months ago

Audio processing by using pytorch 1D convolution network

Python933mit

2 months ago

1d-convolutionaudio-processingcqt-spectrogram

Rust bindings for the C++ api of PyTorch.

Rust3687apache-2.0

2 months ago

deep-learningmachine-learningneural-network

Pytorch Lightning code guideline for conferences

Python1194apache-2.0

7 months ago

A PyTorch Library for Accelerating 3D Deep Learning Research

Python4206apache-2.0

last month

3d-deep-learningartificial-intelligencecamera-api

An Easy To Use PyTorch Computer Vision Library

Python51apache-2.0

9 months ago

computer-visiondeep-learningpytorch

Graph Neural Network Library for PyTorch

Python19309mit

3 months ago

deep-learninggeometric-deep-learninggraph-convolutional-networks

Visual tracking library based on PyTorch.

Python3075gpl-3.0

last month

computer-visionmachine-learningtracking

Bayesian optimization in PyTorch

Jupyter Notebook2847mit

4 months ago

Suite of tools for deploying and training deep learning models using the JVM. Hi

Java13390apache-2.0

last month

artificial-intelligenceclojuredeeplearning

Code Repository for Machine Learning with PyTorch and Scikit-Learn

Jupyter Notebook2839mit

24 days ago

deep-learningmachine-learningneural-networks

Simulation of spiking neural networks (SNNs) using PyTorch.

Python1418agpl-3.0

8 days ago

dynamicgpu-computingmachine-learning

A scikit-learn compatible neural network library that wraps PyTorch

Jupyter Notebook5576bsd-3-clause

2 months ago

hacktoberfesthuggingfacemachine-learning

Build fully-functioning computer vision models with PyTorch

Python607mit

9 months ago

computer-visionfaster-rcnnmachine-learning

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

Python94bsd-3-clause

24 days ago

clinical-data-warehousedeep-learningfast

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

Python7746mit

2 months ago

chainercupydeep-learning

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD

Python690apache-2.0

7 months ago

biologychemistrydeep-chemistry

A simplified framework and utilities for PyTorch

Python556lgpl-3.0

9 months ago

data-sciencedeep-learningkeras

Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, Ba

Jupyter Notebook4867apache-2.0

27 days ago

analytics-zoobigdldistributed-deep-learning

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

C++2491other

2 months ago

caffedeep-learninggpu

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

Python7728mit

5 months ago

machine-learningnumpypytorch

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

Python952mit

7 months ago

3d-object-detectionbevmapcenter

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

Python60agpl-3.0

2 months ago

machine-learningpytorchsnn

A standard framework for modelling Deep Learning Models for tabular data

Python1118mit

3 months ago

deep-learninghacktoberfestmachine-learning

allRank is a framework for training learning-to-rank neural models based on PyTo

Python755apache-2.0

7 months ago

click-modeldeep-learninginformation-retrieval

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

Python13692other

5 months ago

baidudeep-learningdeeplearning

OCaml bindings for PyTorch

OCaml391apache-2.0

11 months ago

artificial-intelligencedeep-learninggpu

MMdnn is a set of tools to help users inter-operate among different deep learnin

Python5771mit

7 months ago

caffecntkcoreml

Deep universal probabilistic programming with Python and PyTorch

Python8238apache-2.0

3 months ago

bayesianbayesian-inferencedeep-learning

Modular reinforcement learning library (on PyTorch and JAX) with support for NVI

Python336mit

5 months ago

deep-learningdeepmindgym

NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

Python17860agpl-3.0

3 months ago

deep-learninghubimage-classification

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Python9850agpl-3.0

3 months ago

deep-learningmachine-learningobject-detection

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Python44373agpl-3.0

3 months ago

coremldeep-learningios

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

Python7578mpl-2.0

last month

aicomputer-visioncv

PyTorch version of Stable Baselines, reliable implementations of reinforcement l

Python7015mit

4 months ago

baselinesgsdegym

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

Python117971apache-2.0

4 months ago

bertdeep-learningflax

Clean PyTorch implementations of imitation and reward learning algorithms

Python1015mit

4 months ago

gymnasiumimitation-learninginverse-reinforcement-learning

Datasets, Transforms and Models specific to Computer Vision

Python14866bsd-3-clause

4 months ago

computer-visionmachine-learning

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

Go2773apache-2.0

last month

data-sciencedeep-learningdistributed-training

gradslam is an open source differentiable dense SLAM library for PyTorch

Python1271mit

8 months ago

3d-reconstructiondeep-learningmachine-learning

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

C++31apache-2.0

5 years ago

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

C++3apache-2.0

2 years ago

Audio generation using diffusion models, in PyTorch.

Python551mit

last year

artificial-intelligenceaudio-generationdeep-learning

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural netw

Python8390mit

last year

artificial-intelligencedeep-learningtext-to-image

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

Cuda4175mit

last year

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

Dynamic Tensor Graph library in Clojure (think PyTorch, DynNet, etc.)

Clojure284epl-1.0

5 years ago

Reinforcement Learning in PyTorch

Python2181mit

3 years ago

Python package facilitating the use of Bayesian Deep Learning methods with Varia

Python351mit

5 years ago

bayesian-inferencedeep-learning

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

Python404

2 years ago

drqapytorchsquad

Reinforcement learning library(framework) designed for PyTorch, implements DQN,

Python376mit

3 years ago

a3c-pytorchddpgdeep-learning

A minimalistic codebase for training NLG models from HuggingFace Transformers us

Python3apache-2.0

last year

bartgenerationgpt-2

Deep recommender models using PyTorch.

Python2919mit

last year

deep-learninglearning-to-rankmachine-learning

PyTorch implementation of DeepDream algorithm

Python130mit

3 years ago

computer-visiondeep-dreamdeepdream

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

Scala47agpl-3.0

last year

array-programmingdottymachine-learning

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

Python282mit

4 years ago

captchacnn-kerasmachine-learning

Probabilistic Programming and Statistical Inference in PyTorch

Python109mit

7 years ago

probabilitypytorchstatistics

Fast, modular reference implementation and easy training of Semantic Segmentatio

Python1399mit

4 years ago

A natural language modeling framework based on PyTorch

Python6356other

2 years ago

PyTorch to Keras model convertor

Python847mit

last year

converterdartsdeep-learning

Pytorch framework for doing deep learning on point clouds.

190other

2 years ago

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

Python456apache-2.0

last year

accuracyarchitectureimagenet

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

Python13apache-2.0

3 years ago

PyTorch implementation of neural style transfer algorithm

Python807mit

2 years ago

deep-learningdeep-styleneural-style

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

Python1204mit

2 years ago

a2ca3cbenchmark

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

Shell46apache-2.0

5 years ago

Graph to sequence implemented in Pytorch combining Graph convolutional networks

Python152mit

5 years ago

This repository contains the official PyTorch implementation of GeoDA algorithm.

Python30

3 years ago