Repository Index

Tensors and Dynamic neural networks in Python with strong GPU acceleration

• neural-network
• autograd
• gpu
• numpy

We write your reusable computer vision tools. 💜

• computer-vision
• image-processing
• python
• yolo

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

• deep-learning
• pytorch
• gpu
• machine-learning

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

• ocr
• deep-learning
• crnn
• pytorch

pix2tex: Using a ViT to convert images of equations into LaTeX code.

• machine-learning
• transformer
• im2latex
• deep-learning

Image augmentation for machine learning experiments.

• image-augmentation
• machine-learning
• augmentation
• deep-learning

Open Machine Learning Compiler Framework

• compiler
• tensor
• deep-learning
• gpu

A collaboration friendly studio for NeRFs

• nerf
• pytorch
• 3d
• 3d-graphics

🐍 Geometric Computer Vision Library for Spatial AI

• computer-vision
• image-processing
• machine-learning
• pytorch

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

• inference
• gpu
• machine-learning
• deep-learning

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

• cloud-computing
• deep-learning
• gpu
• hyperparameter-tuning

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices

• tensorflow
• graph
• darknet
• deep-learning

NVIDIA cuML: GPU-Accelerated Machine Learning

• machine-learning-algorithms
• machine-learning
• cuda
• gpu

Accelerated deep learning R&D

• deep-learning
• reinforcement-learning
• machine-learning
• computer-vision

(IROS 2020, ECCVW 2020) Official Python Implementation for "3D Multi-Object Tracking: A Baseline and New Evaluation Metrics"

• computer-vision
• machine-learning
• robotics
• tracking

DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome

• kmer
• dnabert-model
• kmer-format
• gpu

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

• 3d-object-detection
• 3d-detection
• 3d-deep-learning
• pytorch

Nellie: Automated organelle segmentation, tracking, and hierarchical feature extraction in 2D/3D live-cell microscopy

• biology
• fluorescence
• image-analysis
• image-processing

Accurate segmentation of bacterial microscope images using deep learning synthetically generated image data.

• microscopy
• image-processing
• segmentation
• biology