PyTorch

Evoevo-design/evo

A biological foundation model for long-context DNA sequence modeling and design from single nucleotides to whole genomes.

torchtypingpatrick-kidger/torchtyping

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

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.

whisper-ctranslate2Softcatala/whisper-ctranslate2

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

ProtTransagemagician/ProtTrans

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

packnet-sfmTRI-ML/packnet-sfm

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

TorchXRayVisionmlmed/torchxrayvision

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

DeepDowjankrepl/deepdow

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

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.

Question Generation using hugstransformerspatil-suraj/question_generation

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

Auto Graph LearningTHUMNLab/AutoGL

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

nnAudioKinWaiCheuk/nnAudio

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

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.

SKRLToni-SM/skrl

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

KerasCVkeras-team/keras-cv

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

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.

GitHub repositorydrprojects/superpoint_transformer

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

allRankallegro/allRank

A PyTorch framework for training neural learning-to-rank models with flexible loss functions and scoring 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.

MOSESmolecularsets/moses

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

Platouber-research/plato-research-dialogue-system

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

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.

Anti-UAVZhaoJ9014/Anti-UAV

A benchmark and toolkit for discovering, detecting, recognizing, and tracking UAVs in the wild using RGB and thermal infrared video.

DGL-LifeSciawslabs/dgl-lifesci

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