PyTorch

TResNet: High Performance GPU-Dedicated ArchitecturemrT23/TResNet

A PyTorch implementation of TResNet, a high-performance convolutional neural network architecture optimized for GPU training and inference.

Suite2pMouseLand/suite2p

A complete pipeline for processing two-photon calcium imaging data, including registration, ROI detection, signal extraction, and spike deconvolution.

PathMLDana-Farber-AIOS/pathml

An open-source toolkit for scalable, standardized computational pathology analysis, enabling AI and machine learning on large imaging datasets.

monolocovita-epfl/monoloco

A 3D vision library for monocular and stereo 3D human detection, social distancing, and body orientation estimation from 2D keypoints.

RFUAVkitoweeknd/RFUAV

A benchmark dataset and toolkit for RF-based drone detection and identification using raw IQ data and deep learning models.

sequiturshobrook/sequitur

A PyTorch library for creating and training autoencoders on sequential data (time series, videos, etc.) in just two lines of code.

Huang et al ClinicalBERTkexinhuang12345/clinicalBERT

A BERT-based language model pretrained on clinical notes for predicting hospital readmissions and analyzing medical text.

scFoundationbiomap-research/scFoundation

A 100M-parameter foundation model for single-cell transcriptomics, enabling gene expression enhancement, drug response prediction, and perturbation analysis.

Molecular Transformerpschwllr/MolecularTransformer

A sequence-to-sequence transformer model for predicting chemical reaction pathways (retrosynthesis) with uncertainty calibration.

Whisper-ATYuanGongND/whisper-at

A joint audio tagging and speech recognition model that adds audio event detection to OpenAI Whisper with minimal computational overhead.

YouTube Video TranscriptionArthurFDLR/whisper-youtube

A Google Colab notebook that transcribes YouTube videos using OpenAI's Whisper speech recognition model.

GitHub repositoryedwardzhou130/PolarSeg

A lightweight neural network for near-real-time semantic segmentation of LiDAR point clouds using polar coordinate quantization.

GEARSsnap-stanford/GEARS

A geometric deep learning model that predicts transcriptional outcomes of single and multi-gene perturbations from single-cell RNA-seq data.

MachiniffiX/machin

A PyTorch reinforcement learning library implementing DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, and IMPALA.

CellVit++TIO-IKIM/CellViT

A vision transformer-based deep learning model for automated instance segmentation and classification of cell nuclei in histopathology images.

DrQAhitvoice/DrQA

A PyTorch implementation of the DrQA model for reading comprehension and open-domain question answering.

GPN (Genomic Pre-trained Network)songlab-cal/gpn

A collection of genomic language models for predicting variant effects and evolutionary constraints from DNA sequences.

Torch7prisma-ai/torch2coreml

Convert Torch7 neural network models to Apple CoreML format for deployment on iOS/macOS devices.

DeepLearningNotesAlphaSmartDog/DeepLearningNotes

A collection of open-source machine learning and quantitative analysis models implemented in TensorFlow and PyTorch.

scBERTTencentAILabHealthcare/scBERT

A BERT-based foundation model pretrained on large-scale scRNA-seq data for automated cell type annotation in single-cell analysis.

PyVarInfctallec/pyvarinf

A Python package for training PyTorch neural networks using variational inference for Bayesian deep learning.

UCEsnap-stanford/UCE

A zero-shot foundation model for generating universal embeddings from single-cell gene expression data.

TargetDiffguanjq/targetdiff

Official implementation of a 3D equivariant diffusion model for generating drug-like molecules that bind to specific protein targets and predicting their binding affinity.

AgMLProject-AgML/AgML

A centralized Python framework for agricultural machine learning, providing access to public datasets, benchmarks, pretrained models, and synthetic data generation.

Teleprompterdanielgross/teleprompter

An on-device AI teleprompter that listens to your conversations and suggests charismatic quotes in real-time.

GraphDTAthinng/GraphDTA

GraphDTA predicts drug-target binding affinity using graph neural networks for drug discovery.

Tonicneuromorphs/tonic

A PyTorch Vision-like library for downloading, manipulating, and loading event-based/spike-based neuromorphic datasets.

Mazeenlite-ai/maze

An application-oriented Deep Reinforcement Learning framework for real-world decision problems, covering simulation to deployment.

CNN_kerasskyduy/CNN_keras

A convolutional neural network for CAPTCHA recognition using Keras and PyTorch.

joliGENjolibrain/joliGEN

An integrated framework for training custom generative AI image-to-image models using GANs, Diffusion, and Consistency Models.

Equivariant MLPmfinzi/equivariant-MLP

A JAX library for automatically generating equivariant neural network layers for arbitrary symmetry groups via constraint solving.

PASS: An An ImageNet replacement for self-supervised pretraining without humansyukimasano/PASS

A large-scale image dataset for self-supervised pretraining without humans, designed to reduce privacy concerns.

torch-points3dnicolas-chaulet/torch-points3d

A PyTorch framework for deep learning on point clouds, providing a modular and reproducible foundation for 3D vision tasks.

PyODDSdatamllab/pyodds

An end-to-end Python outlier detection system with database support, automated machine learning, and unified APIs for statistical, ML, and deep learning models.

SecML Malwarepralab/secml_malware

A Python library for creating adversarial attacks against Windows malware detectors to evaluate their robustness.

Caduceuskuleshov-group/caduceus

A bi-directional equivariant transformer for long-range DNA sequence modeling, enabling reverse-complement aware genomic analysis.