OpenMMLab Foundational Library for Training Deep Learning Models
Repository Index
47,133 repositories indexed
121–144 of 183 · Most stars
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
- tensors
- named-tensors
- shape
- pytorch
gradslam is an open source differentiable dense SLAM library for PyTorch
Fast, modular reference implementation and easy training of Semantic Segmentation algorithms in PyTorch.
Official PyTorch inference for Breeze TTS 2
Official PyTorch implementation of VisionHOPE: Visual Backbones as Self-Modifying Learning Systems.
- backbone
- computer-vision
- image-classification
- backbone-networks
Portfolio optimization with deep learning.
- deep-learning
- portfolio-optimization
- finance
- machine-learning
An autoML framework & toolkit for machine learning on graphs.
Audio processing by using pytorch 1D convolution network
- spectrogram-conversion-toolbox
- pytorch
- audio-processing
- preprocessing
Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The PyTorch implementation)
- 3d-object-detection
- lidar-point-cloud
- bevmap
- center
Implementing LLM architectures from scratch in PyTorch
- gemma3
- llama4
- llm
- moe
Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments
- reinforcement-learning
- python
- deep-learning
- gym
allRank is a framework for training learning-to-rank neural models based on PyTorch.
- learning-to-rank
- ndcg
- ranking
- information-retrieval
Pretrained model hub for Keras 3.
Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018
- pytorch
- deep-learning
- generative-adversarial-network
- trajectory-prediction
PyTorch to Keras model convertor
- pytorch
- keras-model-convertor
- keras
- tensorflow
Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs
- point-cloud
- partition
- semantic-segmentation
- pytorch
Don't memorize ML. Understand it from first principles including Inference, Cuda, Deep Learning, Data Science and Reinforcement Learning
- cli
- deep-learning
- education
- from-scratch
Translated者也 (Zheye, "zhe ye", classical Chinese particles) - Zhihu upside-down text Chinese character captcha recognition program
- zhihu
- neural-network
- python
- pytorch
A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)
- deep-learning
- pytorch
- deep-chemistry
- graph-neural-network
OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research
Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.
- deep-learning
- protein-sequences
- protein-structure
- semi-supervised-learning
Build fully-functioning computer vision models with PyTorch
Local AI BGM and sound effects for Claude Code and any MCP client. Stable Audio 3 on Apple Silicon (MLX) and Windows NVIDIA GPUs (PyTorch). No cloud, no API key.
- apple-silicon
- audio-generation
- claude-code
- game-audio