The agent engineering platform

Tensors and Dynamic neural networks in Python with strong GPU acceleration

scikit-learn: machine learning in Python

Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

NumPy & SciPy for GPU

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

Uniform Manifold Approximation and Projection

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

Leveraging BERT and c-TF-IDF to create easily interpretable topics.

Benchmarks of approximate nearest neighbor libraries in Python

A Python implementation of LightFM, a hybrid recommendation algorithm.

Siamese and triplet networks with online pair/triplet mining in PyTorch

Deep recommender models using PyTorch.

Single-cell analysis in Python. Scales to >100M cells.

Code for generating synthetic text images as described in "Synthetic Data for Text Localisation in Natural Images", Ankush Gupta, Andrea Vedaldi, Andrew Zisserman, CVPR 2016.

Reference implementations of MLPerf® training benchmarks

Deep learning for recommender systems

A distributed task scheduler for Dask

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.

PaCMAP: Large-scale Dimension Reduction Technique Preserving Both Global and Local Structure

Approximate Nearest Neighbor Search for Sparse Data in Python!

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 i…

Grounded search engine (i.e. with source reference) based on LLM / ChatGPT / OpenAI API. It supports web search, file content search etc.

(AAAI' 20) A Python Toolbox for Machine Learning Model Combination

Collection of Monte Carlo (MC) and Markov Chain Monte Carlo (MCMC) algorithms applied on simple examples.

Redis Vector Library (RedisVL) -- the AI-native Python client for Redis.

rapids-singlecell: GPU-accelerated framework for scRNA analysis

CuPy fused PyTorch neural networks ops

The TensorFlow reference implementation of 'GEMSEC: Graph Embedding with Self Clustering' (ASONAM 2019).