hanlint

pytorch_geometric

Graph Neural Network Library for PyTorch

TensorRT-LLM

TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.

metaseq

Repo for external large-scale work

composer

Supercharge Your Model Training

distiller

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

neon

Intel® Nervana™ reference deep learning framework committed to best performance on all hardware

llm-foundry

LLM training code for Databricks foundation models

nlp-architect

A model library for exploring state-of-the-art deep learning topologies and techniques for optimizing Natural Language Processing neural networks

bolt

10x faster matrix and vector operations

dbrx

Code examples and resources for DBRX, a large language model developed by Databricks

coach

Reinforcement Learning Coach by Intel AI Lab enables easy experimentation with state of the art Reinforcement Learning algorithms

wer_are_we

Attempt at tracking states of the arts and recent results (bibliography) on speech recognition.

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

TensorNetwork

A library for easy and efficient manipulation of tensor networks.

ngraph

nGraph has moved to OpenVINO

inference

Reference implementations of MLPerf™ inference benchmarks

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

models

Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Intel® Data Center GPUs

diffusion

examples

Fast and flexible reference benchmarks

public_plstm

Phased LSTM

ngraph-python

Original Python version of Intel® Nervana™ Graph

torcheval

A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.

ngraph-bridge

TensorFlow-nGraph bridge

log10

Python client library for managing your LLM data in one place

cerl

vck

Volume Controller for Kubernetes

yahp

hyperparameter management

composer-hydra-template

app.customerdb.com

Voice of customer platform for product teams