Klavis AI: MCP integration platforms that let AI agents use tools reliably at any scale

CUDA Templates and Python DSLs for High-Performance Linear Algebra

Optimized primitives for collective multi-GPU communication

Official inference library for Mistral models

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

Serving multiple LoRA finetuned LLM as one

The simplest, fastest repository for training/finetuning medium-sized GPTs.

Structured state space sequence models

PECOS - Prediction for Enormous and Correlated Spaces

Transformer related optimization, including BERT, GPT

Several SSL methods (Pi model, Mean Teacher) are implemented in pytorch

Data augmentation for NLP

A state-of-the-art semi-supervised method for image recognition

A Survey

State-of-the-Art Embeddings, Retrieval, and Reranking

Open-source AI orchestration framework for building context-engineered, production-ready LLM applications. Design modular pipelines and agent workflows with explicit control over retrieval, routing…

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Hydra is a framework for elegantly configuring complex applications

A Unified Library for Parameter-Efficient and Modular Transfer Learning

Python package built to ease deep learning on graph, on top of existing DL frameworks.

sentence embedding by Smooth Inverse Frequency weighting scheme

DeText: A Deep Neural Text Understanding Framework for Ranking and Classification Tasks

A simple way to calibrate your neural network.

Repository for the tutorial on Sequence-Aware Recommender Systems held at TheWebConf 2019 and ACM RecSys 2018

Deep recommender models using PyTorch.

HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training

AI and Machine Learning with Kubeflow, Amazon EKS, and SageMaker

An industrial deep learning framework for high-dimension sparse data

A toolkit for developing and comparing reinforcement learning algorithms.