# MAX AI kernels API (Mojo)

> The MAX AI kernels library reference (top-level package indexes).

This file contains links to documentation sections following the llmstxt.org standard.

## Table of Contents

- [builtin_kernels](https://docs.modular.com/max/api/kernels/builtin_kernels.md): Mojo package builtin_kernels.builtin_kernels documentation
- [comm](https://docs.modular.com/max/api/kernels/comm.md): Provides communication primitives for GPUs.
- [extensibility](https://docs.modular.com/max/api/kernels/extensibility.md): Surface needed to write GraphCompiler kernel entry points.
- [kernels](https://docs.modular.com/max/api/kernels.md): High-performance compute kernels, operators, and tensor APIs for MAX graphs.
- [kv_cache](https://docs.modular.com/max/api/kernels/kv_cache.md): Contains implementations for several types of key-value caches.
- [linalg](https://docs.modular.com/max/api/kernels/linalg.md): Provides CPU and GPU implementations of linear algebra functions.
- [nn](https://docs.modular.com/max/api/kernels/nn.md): Provides neural network operators for deep learning models.
- [nvml](https://docs.modular.com/max/api/kernels/nvml.md): Implements wrappers around the NVIDIA Management Library (nvml).
- [pipeline](https://docs.modular.com/max/api/kernels/pipeline.md): Generic compile-time software pipeline scheduling framework.
- [quantization](https://docs.modular.com/max/api/kernels/quantization.md): This package contains a set of APIs for quantizing tensor data.
- [shmem](https://docs.modular.com/max/api/kernels/shmem.md): Implements a subset of OpenSHMEM functionality.
- [state_space](https://docs.modular.com/max/api/kernels/state_space.md): State space model kernels for Mamba and related architectures.
- [structured_kernels](https://docs.modular.com/max/api/kernels/structured_kernels.md): Shared GPU kernel primitives for structured kernel architectures.