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.

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

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.

High-performance TensorFlow library for quantitative finance.

cuML - RAPIDS Machine Learning Library

Time series forecasting with PyTorch

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

eBPF Observability - Distributed Tracing and Profiling

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

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

Single C file, Realtime CPU/GPU Profiler with Remote Web Viewer

The Cross Platform Game Engine

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

cuGraph - RAPIDS Graph Analytics Library