Search awesome repositories

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A cross-platform, OpenGL terminal emulator.

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

macOS system monitor in your menu bar

The fastai deep learning library

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

Open3D: A Modern Library for 3D Data Processing

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

Open Machine Learning Compiler Framework

NumPy & SciPy for GPU

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

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

cuDF - GPU DataFrame Library

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

a language for fast, portable data-parallel computation

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

FlashInfer: Kernel Library for LLM Serving

cuML - RAPIDS Machine Learning Library

Time series forecasting with PyTorch

On-device AI across mobile, embedded and edge for PyTorch

🛠 All-in-one web-based IDE specialized for machine learning and data science.

Achieve state of the art inference performance with modern accelerators on Kubernetes

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

cuGraph - RAPIDS Graph Analytics Library

Vendor-agnostic orchestration for training, inference and agentic workloads across NVIDIA, AMD, TPU, and Tenstorrent on clouds, Kubernetes, and bare metal.

ThunderSVM: A Fast SVM Library on GPUs and CPUs

NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale datasets used to train deep learning based recommender systems.

ThunderGBM: Fast GBDTs and Random Forests on GPUs

Kubernetes operator for local LLM inference with llama.cpp, vLLM, TGI, and mlx-server — multi-GPU NVIDIA + Apple Silicon Metal, autoscaling, air-gapped, production-ready