Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark
A benchmark framework evaluating accuracy-efficiency tradeoffs of time series foundation models on consumer GPUs (6GB VRAM).
| Model | Parameters | Type | Expected VRAM |
|---|---|---|---|
| XGBoost | ~1M | Tree-based | 0 GB (CPU) |
| ARIMA | - | Statistical | 0 GB (CPU) |
| DLinear | 98.5K | Efficient DL | ~0.1 GB |
| PatchTST | 1.2M | Efficient DL | ~0.5 GB |
| Chronos-Tiny | 8M | Foundation | ~1 GB |
| Chronos-Mini | 20M | Foundation | ~2 GB |
| Chronos-Small | 46M | Foundation | ~3 GB |
| Moirai-Small | 14M | Foundation | ~1.5 GB |
| Dataset | Domain | Frequency | Series |
|---|---|---|---|
| ETTh1 | Energy | Hourly | 7 |
| Exchange | Finance | Daily | 8 |
| Traffic | Transport | Hourly | 862 |
| Weather | Climate | 10-min | 21 |
# 1. Create virtual environment
python -m venv venv
source venv/bin/activate # Linux/Mac
# OR: venv\Scripts\activate # Windows
# 2. Install PyTorch with CUDA first
pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cu118
# 3. Install dependencies
pip install -r requirements.txt
# 4. Download datasets
python -m src.data.download
# 5. Run benchmark
python experiments/run_baselines.py --model xgboost --dataset etth1 --runs 5
python experiments/run_foundation.py --model chronos --size small --dataset etth1
edge_inference_benchmark/
βββ configs/ # Model and dataset configurations
βββ src/ # Source code
β βββ data/ # Data loading
β βββ models/ # Model implementations
β βββ evaluation/ # Metrics and efficiency
β βββ visualization/ # Plotting utilities
βββ experiments/ # Experiment scripts
βββ results/
β βββ figures/ # Visualization outputs
β βββ benchmark_summary.json # All metrics summary
βββ requirements.txt


Tested on:
@article{author2026edge,
title={Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark},
author={Author, Your Name},
journal={Transactions on Machine Learning Research},
year={2026}
}
MIT License - see LICENSE for details.
4 commits
Python
100.0%
Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark
A benchmark framework evaluating accuracy-efficiency tradeoffs of time series foundation models on consumer GPUs (6GB VRAM).
| Model | Parameters | Type | Expected VRAM |
|---|---|---|---|
| XGBoost | ~1M | Tree-based | 0 GB (CPU) |
| ARIMA | - | Statistical | 0 GB (CPU) |
| DLinear | 98.5K | Efficient DL | ~0.1 GB |
| PatchTST | 1.2M | Efficient DL | ~0.5 GB |
| Chronos-Tiny | 8M | Foundation | ~1 GB |
| Chronos-Mini | 20M | Foundation | ~2 GB |
| Chronos-Small | 46M | Foundation | ~3 GB |
| Moirai-Small | 14M | Foundation | ~1.5 GB |
| Dataset | Domain | Frequency | Series |
|---|---|---|---|
| ETTh1 | Energy | Hourly | 7 |
| Exchange | Finance | Daily | 8 |
| Traffic | Transport | Hourly | 862 |
| Weather | Climate | 10-min | 21 |
# 1. Create virtual environment
python -m venv venv
source venv/bin/activate # Linux/Mac
# OR: venv\Scripts\activate # Windows
# 2. Install PyTorch with CUDA first
pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cu118
# 3. Install dependencies
pip install -r requirements.txt
# 4. Download datasets
python -m src.data.download
# 5. Run benchmark
python experiments/run_baselines.py --model xgboost --dataset etth1 --runs 5
python experiments/run_foundation.py --model chronos --size small --dataset etth1
edge_inference_benchmark/
βββ configs/ # Model and dataset configurations
βββ src/ # Source code
β βββ data/ # Data loading
β βββ models/ # Model implementations
β βββ evaluation/ # Metrics and efficiency
β βββ visualization/ # Plotting utilities
βββ experiments/ # Experiment scripts
βββ results/
β βββ figures/ # Visualization outputs
β βββ benchmark_summary.json # All metrics summary
βββ requirements.txt


Tested on:
@article{author2026edge,
title={Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark},
author={Author, Your Name},
journal={Transactions on Machine Learning Research},
year={2026}
}
MIT License - see LICENSE for details.
4 commits
Python
100.0%