XTREME is a benchmark for the evaluation of the cross-lingual generalization ability of pre-trained multilingual models that covers 40 typologically diverse languages and includes nine tasks.

MTEB: Massive Text Embedding Benchmark

An implementation of a deep learning recommendation model (DLRM)

Model interpretability and understanding for PyTorch

Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining the behavior of Transformer-based language models (like GPT2, B…

Efficient few-shot learning with Sentence Transformers

The Learning Interpretability Tool: Interactively analyze ML models to understand their behavior in an extensible and framework agnostic interface.

Lightning-fast serving engine for any AI model of any size. Flexible. Easy. Enterprise-scale.

Toolbox of models, callbacks, and datasets for AI/ML researchers.

A simple way to calibrate your neural network.

A high-throughput and memory-efficient inference and serving engine for LLMs

cuDF - GPU DataFrame Library

cuML - RAPIDS Machine Learning Library

Automated support ticket classification using NLP and ML algorithms.

Time series forecasting with PyTorch