A highly efficient implementation of Gaussian Processes in PyTorch

Generative AI reference workflows optimized for accelerated infrastructure and microservice architecture.

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

Sionna: An Open-Source Library for Next-Generation Physical Layer Research

NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.

GPU-accelerated Deep Learning on Windows 10 native

A full pipeline AutoML tool for tabular data

PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.

Stretching GPU performance for GEMMs and tensor contractions.

Efficient Spiking Neural Network framework, built on top of PyTorch for GPU acceleration

Distributed GPU-Accelerated Framework for Evolutionary Computation. Comprehensive Library of Evolutionary Algorithms & Benchmark Problems.

TOD: GPU-accelerated Outlier Detection via Tensor Operations

Kafka-ML: connecting the data stream with ML/AI frameworks (now TensorFlow and PyTorch!)

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

Solvers for NP-hard and NP-complete problems with an emphasis on high-performance GPU computing.

An easy way to use anime4k in python

Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed computing

Data Parallel Extension for NumPy

Model Predictive Path Integral Control (MPPI) with PyTorch

Package xrt (XRayTracer) is a python software library for ray tracing and wave propagation in x-ray regime. It is primarily meant for modeling synchrotron sources, beamlines and beamline elements.