Skip to content

Repository files navigation

compute

Go to Installation Guide

Go to Inference Cookbook Recipes

Go to Benchmark Data for Decode Throughput

Go to Included Skills

Go to Additional Tools


Introduction

This repo provides a quick & simple way to stage an Ubuntu / RHEL NVIDIA node for ML workloads, including serving inference cookbook recipes for Large Language Models & Visual-Language Models, while providing the tools to build new inference cookbook recipes. Setup and stage a bare-metal server, virtual machine, or rental cloud provider node (Verda, Massed Compute, Prime Intellect, etc). AMD and additional hardware vendor support is in the works.

An installation guide is provided with steps to clone this repo and run setup.sh to display the above terminal interface. All options excluding the Coding CLIs are selected by default with the ability to choose or omit packages / services, single or multiple CUDA versions, Python, Astral UV, & coding CLIs.

You can also select the minimal installation of basic Linux essentials and core build dependencies. This option completely bypasses the installation of additional packages / services, CUDA, Python, Astral UV, & coding CLIs.

Installing CUDA, Node.js 24, or Python through this guide replaces existing defaults, using Node Version Manager (NVM) for managing Node.js 24 & Simple Python Version Management (Pyenv) for managing Python. CUDA drivers are installed and/or updated to the latest version within a CUDA installation.

Inference cookbook recipes are also included for a variety of the popular Large Language Model & Visual-Language model companies. These exist as bash scripts and automatically create new Python environments within the $HOME directory upon execution, while installing pinned versions of either SGLang or vLLM + any additional packages necessary for a proper inference deployment. Additionally each script has the ability customize specific flags based off your CUDA architecture target. Benchmark data for decode throughput per model, GPU hardware, and GPU quantity are also included.

New recipes can be created using the provided skills within this repo, added to your coding CLI of choice, and building them based off the existing recipes.


Installation Guide:

setup.sh runs up to 4 installer files:

installers/01_install_dependencies.sh
installers/02_install_cuda.sh
installers/03_install_python.sh
installers/04_install_coding_clis.sh

The /workspace is the default repo directory used in this guide, yet you're at liberty to choose any other path for the installation.

Clone the Repo onto a GPU Instance with a Root user:

mkdir -p /workspace
cd /workspace
git clone https://github.com/yoniq-org/compute.git
cd compute
./setup.sh

Clone the Repo onto a GPU Instance with an Ubuntu (non-Root) user:

sudo mkdir -p /workspace
sudo chown -R ubuntu:ubuntu /workspace
cd /workspace
git clone https://github.com/yoniq-org/compute.git
cd compute
./setup.sh

The below are each package / service provided across the installers.

Basic Linux essentials (required):

  • curl, wget, zip, unzip, less, vim, nano, tmux, git, git-lfs, gh, htop, nvtop, ripgrep, shellcheck, bubblewrap, ffmpeg

Core build dependencies for ML and Python packages (required):

  • build-essential, gcc, g++, make, cmake, pkg-config, protobuf-compiler, libclang-dev, numactl, libnuma-dev, libhwloc-dev, libssl-dev, libffi-dev, liblzma-dev, libbz2-dev, libreadline-dev, libsqlite3-dev, libncurses-dev, zlib1g-dev

Additional services (optional):

  • Docker, Node.js 24 (NVM managed), pnpm, Bun, Go, Rust, Zig, Neovim, Tmux

CUDA (optional):

  • CUDA 13.2 is selected by default. For a custom install you can type either any CUDA version number, or up to 10 version numbers with the 1st number in the list set as the default system-wide CUDA version

Pyenv, Pyenv managed Python, & Astral UV (optional):

  • Python 3.12.14 is selected by default. For a custom install you can type any other version of Python in full major.mino.macro format. Pyenv is first installed following the desired Python version. Astral UV is provided as an installation option if selected.

Coding CLIs (optional):

  • Arcee nac, Claude Code, DeepSeek Harness, Gemini CLI, Grok Build, Kimi Code, Meta Muse Code, MiMo Code, MiniMax Code, OMP, OpenAI Codex, OpenCode, Pi, Prime Intellect Agent, Qwen Code

Inference Cookbook Recipes:

Inference cookbook recipes are provided as bash scripts, validated against 1x, 2x, 4x, and 8x NVIDIA H200s, with future support for sm100, sm103, sm120, & sm121 architecture NVIDIA GPUs.

The currently supported companies are:

  • Allen Institute for AI, Arcee AI, Cohere, Datalab, DeepSeek, Dots Studio, Google, IBM Granite, Inclusion Ai, Inco AI, Inferact, Intel, Liquid AI, Meta, Microsoft, MiniMax, Mistral AI, Moonshot AI, Nanbeige, Nex-AGI, NVIDIA, OpenAI, Paradigma, Prime Intellect, Poolside, Qwen, RadixArk, Red Hat AI, StepFun, Tencent, Thinking Machines Lab, Xiaomi, Yandex, Z Lab, Z.ai, Zyphra

All scripts follow a singular format, allowing easy replicability for newer models & hardware architecture. Specific flags are also defined by CUDA architecture target, allowing the same script to run on multiple hardware, with a singular architecture per launch.

Launch each script to automatically detect your GPU's CUDA architecture and run inference. For heterogenous GPU setups, provide CUDA_VISIBLE_DEVICES= followed by your GPU(s) of choice as a script argument to define your target CUDA architecture from the first GPU in the list.

Launch examples:

  • ./vllm_Qwen_Qwen3.8-27B.sh
  • CUDA_VISIBLE_DEVICES=1 ./vllm_Qwen_Qwen3.8-27B.sh

Below are the CUDA architecture specific flags per cookbook recipe script:

Script Envs SGLang Flag vLLM Flag
BACKEND_ATTENTION_SM* --attention-backend --attention-backend
BACKEND_FP8_GEMM_SM* --fp8-gemm-backend --linear-backend
BACKEND_FP4_GEMM_SM* --fp4-gemm-backend --linear-backend
BACKEND_MOE_RUNNER_SM* --moe-runner-backend --moe-backend
GPU_MEM_UTIL_VALUE_SM* --mem-fraction-static --gpu-mem-util
TENSOR_PARALLEL_SIZE_SM* --tp --tensor-parallel-size

Benchmark Data for Decode Throughput:

Under each recipe directory is an llm-inference-bench directory containing decode throughput benchmark data, using a tool created by the wonderful crew over at Local Inference Lab. Each benchmark file is categorized by model script, GPU type, and GPU quantity.

Local Inference Lab's contact info and their LLM Inference Bench repo can be found at:

Below is an example decode throughput data for DeepSeek-V4-Flash-0731 running on 2x NVIDIA H200 NVL GPUs:

llm-inference-bench-example


Included Skills:


Additional Tools:

Change Between Python Environments:

source ./launch_env.sh

Install any selection of open-weights models or input your desired repo:

./model_download.sh or example with repo: ./model_download.sh Qwen/Qwen3.8-27B

Check for any model snapshot / blob updates for existing repos:

./check_model_updates.sh

About

No description, website, or topics provided.

Resources

Contributing

Stars

53 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages