Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions .zenodo.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"title": "batch-probe: GPU memory probing and thermal-aware CPU thread control",
"upload_type": "software",
"description": "Probes the largest OOM-safe GPU batch size (binary search with OOM recovery) and runs a Kalman-filtered, thermal-aware controller for CPU thread / GPU load to keep a node within its thermal envelope. Works with PyTorch, CuPy, JAX, or any GPU framework.",
"creators": [
{"name": "Bond, Andrew H.", "affiliation": "San Jose State University", "orcid": "0009-0003-2599-6158"}
],
"license": {"id": "MIT"},
"keywords": ["pytorch", "cupy", "jax", "gpu", "memory", "batch-size", "oom", "cuda", "thermal", "threads"]
}
26 changes: 26 additions & 0 deletions CITATION.cff
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
cff-version: 1.2.0
message: "If you use batch-probe in your research, please cite it as below."
title: "batch-probe: GPU memory probing and thermal-aware CPU thread control"
type: software
authors:
- family-names: Bond
given-names: Andrew H.
email: andrew.bond@sjsu.edu
affiliation: San Jose State University
orcid: "https://orcid.org/0009-0003-2599-6158"
version: "0.5.0"
date-released: "2026-06-30"
license: MIT
url: "https://github.com/ahb-sjsu/batch-probe"
repository-code: "https://github.com/ahb-sjsu/batch-probe"
keywords:
- pytorch
- cupy
- jax
- gpu
- memory
- batch-size
- oom
- cuda
- thermal
- threads
Loading