> ## Documentation Index
> Fetch the complete documentation index at: https://docs.coreweave.com/llms.txt
> Use this file to discover all available pages before exploring further.

# March 31, 2026 - New marimo notebook tutorials: JAX and TensorRT-LLM

> New tutorials for running JAX training and TensorRT-LLM inference in interactive marimo notebooks on CKS

Two new tutorials are available for running GPU workloads in interactive [marimo](https://marimo.io/) notebooks on CKS.

### JAX training with marimo

The [JAX tutorial](/products/cks/tutorials/marimo-notebooks/jax) deploys a JAX example notebook to CKS with a single CLI command. The notebook trains a small identity network using Flax and Optax, and streams a live loss chart to the browser as training progresses. The example uses the standard `ghcr.io/marimo-team/marimo:latest` image, no custom Dockerfile needed. Dependencies including `jax[cuda12]` are pinned in a [PEP 723](https://peps.python.org/pep-0723/) header embedded in the notebook file, so the environment is recreated exactly on every pod start.

### TensorRT-LLM inference with marimo

The [TensorRT-LLM tutorial](/products/cks/tutorials/marimo-notebooks/tensorrt) runs NVIDIA TensorRT-LLM inference inside a marimo notebook on CKS using the official NVIDIA Triton NGC container. The notebook includes an interactive model picker and prompt selector. Supported models include TinyLlama 1.1B, Phi-3.5-mini, Mistral 7B, Llama-3.1 8B FP8, and Minitron 8B. The tutorial supports both the `kubectl-marimo` CLI plugin and a plain YAML manifest.

Both tutorials require the [marimo operator](/products/cks/tutorials/marimo-notebooks) installed on your cluster.