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

Build and run Docker containers leveraging NVIDIA GPUs

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

Open Machine Learning Compiler Framework

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

NumPy & SciPy for GPU

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

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

cuDF - GPU DataFrame Library

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.

Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, DeepSeek, Mixtral, Gemma, Phi, MiniCPM, Qwen-VL, MiniCPM-V, etc.) on Intel XPU (e.g., local PC with iGPU and NPU, discrete GPU such as Arc, Flex and Max); seamlessly integrate with llama.cpp, Ollama, HuggingFace, LangChain, LlamaIndex, vLLM, DeepSpeed, Axolotl, etc.

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

FlashInfer: Kernel Library for LLM Serving

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

Lemonade helps users discover and run local AI apps by serving optimized LLMs right from their own GPUs and NPUs. Join our discord: https://discord.gg/5xXzkMu8Zk

High-performance TensorFlow library for quantitative finance.

NVIDIA cuML: GPU-Accelerated Machine Learning

cuML - RAPIDS Machine Learning Library

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

Time series forecasting with PyTorch

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

eBPF Observability - Distributed Tracing and Profiling

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