World's first general purpose 3D object detection codebse.
- nuscenes
- 3d-object-detection
- point-cloud
- object-detection
Repository Index
47,133 repositories indexed
169–192 of 245 · Most stars
OpenMMLab Foundational Library for Training Deep Learning Models
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
- tensors
- named-tensors
- shape
- pytorch
Official PyTorch inference for Breeze TTS 2
An open source DevOps tool from the CNCF for packaging and versioning AI/ML models, datasets, code, and configuration into an OCI Artifact.
gradslam is an open source differentiable dense SLAM library for PyTorch
Official PyTorch implementation of VisionHOPE: Visual Backbones as Self-Modifying Learning Systems.
- backbone
- computer-vision
- image-classification
- backbone-networks
Fast, modular reference implementation and easy training of Semantic Segmentation algorithms in PyTorch.
A collection of computer vision pre-trained models.
A JavaScript library like PyTorch, with GPU acceleration.
- automatic-differentiation
- deep-learning
- javascript-library
- neural-networks
TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.
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.
A uniform interface to run deep learning models from multiple frameworks
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
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
Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs
- point-cloud
- partition
- semantic-segmentation
- pytorch