🧠

LLM Fine-Tuning Lab

GCP Compute Engine with NVIDIA A100 80 GB

Overview

The LLM Fine-Tuning Lab is the crown jewel of our lab infrastructure. Each student gets a dedicated NVIDIA A100 with 80 GB of VRAM — the highest-memory GPU available on GCP — giving you enough room to fine-tune 7B parameter models with full precision or run QLoRA on models up to 70B parameters. This is the same hardware that AI companies use to create their custom language models. The 1 TB SSD ensures fast model loading — downloading a 70B model from Hugging Face takes minutes, not hours. Pre-installed tooling includes Unsloth for 2x faster fine-tuning, Axolotl for config-driven training, and Flash Attention 2 for memory-efficient attention computation.

What You'll Do in This Lab

  • Fine-tune Llama 3, Mistral, and Gemma models on domain-specific datasets
  • Apply LoRA and QLoRA adapters for parameter-efficient fine-tuning
  • Fine-tune 70B parameter models in 4-bit quantization on a single GPU
  • Create instruction-tuned chat models from base models
  • Implement DPO alignment to steer model behavior with preference data
  • Merge LoRA adapters back into base models for deployment

Lab Workflow

1

Request

LLM fine-tuning labs are scheduled sessions (3-6 hours). Book from your course dashboard. Base models are pre-cached on the instance SSD.

2

Launch

Your A100 80GB instance launches with the selected base model already loaded on the 1 TB SSD. No download wait time.

3

Prepare Data

Format your training dataset using the provided templates — instruction/response pairs, chat format, or preference pairs for DPO.

4

Configure

Set training hyperparameters: LoRA rank, learning rate, batch size, quantization settings. Use provided config templates or customize.

5

Fine-Tune

Launch training with Unsloth or Axolotl. Monitor loss curves, learning rate schedule, and GPU memory in real-time via W&B.

6

Evaluate

Run your fine-tuned model on evaluation prompts. Compare outputs against the base model. Run automated quality benchmarks.

7

Export

Merge LoRA adapters, save the final model, and push to Cloud Storage or Hugging Face Hub for use in serving labs.

Hardware & Environment

Machine Type a2-ultragpu-1g (12 vCPU, 170 GB RAM)
GPU 1x NVIDIA A100 80 GB HBM2e (Ampere architecture)
GPU Memory Bandwidth 2 TB/s HBM2e
Storage 1 TB NVMe SSD (pre-cached with popular base models)
Max Model Size 7B full precision, 13B in 8-bit, 70B in 4-bit (QLoRA)
Session Length 3-6 hour scheduled slots

Pre-installed Tools

Hugging Face Transformers + PEFT Unsloth Axolotl bitsandbytes (4-bit/8-bit quantization) Weights & Biases Flash Attention 2

Frequently asked questions about this lab

What is the LLM Fine-Tuning Lab? +
High-memory GPU environment purpose-built for fine-tuning large language models. Supports full fine-tuning of 7B+ parameter models and parameter-efficient methods (LoRA, QLoRA) for larger models up to 70B.
Which courses use this lab? +
This lab is included in: NLP & Large Language Model Engineering, Generative AI & Prompt Engineering.
What hardware does this lab run on? +
GCP Compute Engine with NVIDIA A100 80 GB. Machine Type: a2-ultragpu-1g (12 vCPU, 170 GB RAM); GPU: 1x NVIDIA A100 80 GB HBM2e (Ampere architecture); GPU Memory Bandwidth: 2 TB/s HBM2e; Storage: 1 TB NVMe SSD (pre-cached with popular base models).
What software comes pre-installed? +
Comes pre-loaded with Hugging Face Transformers + PEFT, Unsloth, Axolotl, bitsandbytes (4-bit/8-bit quantization), Weights & Biases, Flash Attention 2. No local installs or dependency setup required — open your browser and start working.
Can I bring my own datasets and code into this lab? +
Yes. Datasets can be uploaded directly or synced from Google Cloud Storage. Notebooks and source files have built-in Git integration so you can push work to your own GitHub or GitLab repos.
Do I need to enroll in a course to use this lab? +
Yes. Lab environments are provisioned per-student as part of an AI Labs course enrollment. Browse the courses linked above to find programs that include this lab.

Related labs

Other AI Labs environments students typically use alongside this one.

Ready to Try This Lab?

Enroll in a course that uses this lab, or visit our Houston center for a hands-on demo.