PyTorch

torchtypingpatrick-kidger/torchtyping

Type annotations and runtime checking for PyTorch tensor shape, dtype, layout, and names.

LightlyTrainlightly-ai/lightly-train

An all-in-one framework for training state-of-the-art computer vision models, covering pretraining, fine-tuning, and distillation.

MMEngineopen-mmlab/mmengine

A foundational PyTorch library for training deep learning models, serving as the core engine for the OpenMMLab ecosystem.

gradslamgradslam/gradslam

An open-source differentiable dense SLAM library for PyTorch, enabling gradient flow from map outputs to sensor inputs.

TorchSegycszen/TorchSeg

A fast, modular PyTorch reference implementation for training and evaluating semantic segmentation models.

PersonaeCeruleanacg/Personae

A repository implementing Deep Reinforcement Learning and Supervised Learning methods with a simulated financial market environment for quantitative trading.

SLM Labkengz/SLM-Lab

A modular deep reinforcement learning framework in PyTorch for research and application, featuring ready-to-use algorithms and reproducible experiments.

ProtTransagemagician/ProtTrans

State-of-the-art pre-trained transformer language models for protein sequences, enabling tasks like structure prediction and function annotation.

whisper-ctranslate2Softcatala/whisper-ctranslate2

A faster, memory-efficient command-line client for OpenAI's Whisper speech recognition, powered by CTranslate2.

packnet-sfmTRI-ML/packnet-sfm

A PyTorch implementation of self-supervised monocular depth estimation using 3D packing for high-resolution, real-time depth prediction.

DeepPurposekexinhuang12345/DeepPurpose

A deep learning library for drug-target interaction, drug property, protein-protein interaction, drug-drug interaction, and protein function prediction in bioinformatics.

PPLMuber-research/PPLM

A method to steer topic and attributes of GPT-2 language models without fine-tuning, enabling controlled text generation.

TorchXRayVisionmlmed/torchxrayvision

An open-source library providing chest X-ray datasets, pre-trained models, and tools for medical imaging research and analysis.

Question Generation using hugstransformerspatil-suraj/question_generation

An open-source study on neural question generation using transformers, providing simplified training and inference pipelines.

DeepDowjankrepl/deepdow

A Python framework for portfolio optimization using deep learning to allocate investment weights in a single forward pass.

Auto Graph LearningTHUMNLab/AutoGL

An autoML framework and toolkit for automating machine learning tasks on graph-structured data.

Super-Fast-Accurate-3D-Object-Detectionmaudzung/Super-Fast-Accurate-3D-Object-Detection

A PyTorch implementation for super fast and accurate 3D object detection using LiDAR point clouds, featuring an anchor-free approach.

nnAudioKinWaiCheuk/nnAudio

An audio processing toolbox using PyTorch 1D convolutional neural networks for on-the-fly spectrogram generation with trainable kernels.

KerasCVkeras-team/keras-cv

A library of modular computer vision components built on Keras 3, supporting TensorFlow, JAX, and PyTorch backends.

SKRLToni-SM/skrl

A modular Python library for Reinforcement Learning with support for PyTorch, JAX, NVIDIA Warp, and multiple environment interfaces.

CornacPreferredAI/cornac

A comparative Python framework for building, evaluating, and deploying multimodal recommender systems with auxiliary data.

lidar-bonnetalPRBonn/lidar-bonnetal

A framework for semantic and instance segmentation of LiDAR point clouds using range images, designed for autonomous driving applications.

allRankallegro/allRank

A PyTorch framework for training neural learning-to-rank models with flexible loss functions and scoring architectures.

GitHub repositorydrprojects/superpoint_transformer

A PyTorch framework for efficient 3D semantic and panoptic segmentation using superpoint-based transformer architectures.

OpenMetricLearningOML-Team/open-metric-learning

A PyTorch-based framework for training and validating models that produce high-quality embeddings for metric learning and retrieval tasks.

ai-imu-drmbrossar/ai-imu-dr

A deep learning-enhanced Kalman filter for accurate vehicle dead reckoning using only an IMU sensor.

KerasNLPkeras-team/keras-nlp

A pretrained modeling library for Keras 3 offering simple, flexible, and fast access to models for text, image, and audio tasks.

Platouber-research/plato-research-dialogue-system

A flexible Python framework for developing, training, and evaluating conversational AI agents in single or multi-agent environments.

MOSESmolecularsets/moses

A benchmarking platform for molecular generation models, providing datasets, implementations, and evaluation metrics for drug discovery research.

GibsonEnvStanfordVL/GibsonEnv

A virtual environment simulator for training embodied AI agents with real-world perception and physics, featuring domain transfer to real robots.

sganagrimgupta92/sgan

A PyTorch implementation of Social GAN for predicting socially acceptable human trajectories using generative adversarial networks.

neural-style-ptProGamerGov/neural-style-pt

A PyTorch implementation of neural style transfer, combining the content of one image with the artistic style of another.

LLVIPbupt-ai-cz/LLVIP

A visible-infrared paired dataset for low-light vision tasks like pedestrian detection, image fusion, and image-to-image translation.

anchormarcotcr/anchor

A model-agnostic method for generating high-precision rule-based explanations for black-box classifier predictions.

GitHub repositoryloicland/superpoint_graph

A PyTorch framework for semantic segmentation of large 3D point clouds using superpoint graphs.

DGL-LifeSciawslabs/dgl-lifesci

A Python package for applying graph neural networks to molecular graphs and biological networks in life science research.