Repository Index

Tensors and Dynamic neural networks in Python with strong GPU acceleration

• neural-network
• autograd
• gpu
• numpy

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

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

Productive, portable, and performant GPU programming in Python.

• taichi
• differentiable-programming
• gpu-programming
• sparse-computation

The fastai deep learning library

• deep-learning
• machine-learning
• pytorch
• python

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...

• java
• gpu
• deeplearning
• neural-nets

Open3D: A Modern Library for 3D Data Processing

• mesh-processing
• computer-graphics
• opengl
• cpp

Open Machine Learning Compiler Framework

• compiler
• tensor
• deep-learning
• gpu

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

• python
• profiling
• performance-analysis
• cpu-profiling

NumPy & SciPy for GPU

• cuda
• cudnn
• cublas
• cusolver

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

CUDA Templates and Python DSLs for High-Performance Linear Algebra

• cuda
• deep-learning
• deep-learning-library
• cpp

cuDF - GPU DataFrame Library

• gpu
• rapids
• cudf
• arrow

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.

• machine-learning
• decision-trees
• gradient-boosting
• gbm

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.

• h2o
• machine-learning
• data-science
• deep-learning

《深入理解 AI Infra：量化分析与系统设计》（李博杰 著）开源书稿：从硬件约束和模型架构出发，量化推导 LLM 推理与训练系统设计。含全书正文、PDF、配套计算工具与实验

• accelerator
• ai-infra
• ai-infrastructure
• book

High-performance TensorFlow library for quantitative finance.

• tensorflow
• quantitative-finance
• python
• finance

NVIDIA cuML: GPU-Accelerated Machine Learning

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

Up to 100x faster strings for C, C++, CUDA, Python, Rust, Swift, JS, & Go, leveraging NEON, AVX2, AVX-512, SVE, GPGPU, & SWAR to accelerate search, hashing, sorting, edit distances, sketches, and memory ops 🦖

• simd
• dataset
• string
• string-manipulation

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

• machine-learning
• deep-learning
• data-science
• docker

PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer

• graph
• visualization
• gpu
• graphistry

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

• machine-learning
• python
• statistics
• scikit-learn

A collection of GICP-based fast point cloud registration algorithms

• point-cloud
• pcl
• gicp
• icp

An efficient C++20 GPU numerical computing library with Python-like syntax

• cuda
• gpu
• gpgpu
• hpc