Compute on a budget
You can do real research without a big GPU cluster. Start small, use free and academic resources wisely, write efficient code, and know when you truly need more.
In 60 seconds
Compute on a budget
You can do real research without a big GPU cluster. Start small, use free and academic resources wisely, write efficient code, and know when you truly need more.
Small models first
- Small models: many open models come in several sizes. Use the smallest one that can show the effect you care about.
- Subsets: debug on 1% of the data. Only use the full dataset when the pipeline works.
- Short runs: check that loss goes down for a few hundred steps before launching a long run.
- Scaling checks: try two or three small sizes. If the trend is steady, you have evidence about what may happen at larger scale. It is a hint, not a guarantee.
- Adapt instead of train: fine-tuning with LoRA or QLoRA, or using a model as it is, is far cheaper than training from scratch.
Where to find compute
| Source | What it is | Things to know |
|---|---|---|
| Your own laptop | CPU, sometimes a small GPU | Good for small models, data work, evaluation and debugging. Free and always available. |
| Free notebook tiers | Some cloud notebook services give free, limited GPU time | Limits, GPU types and rules change often. Sessions can stop without warning, so save checkpoints. Check the current terms. |
| University computers | Shared GPU servers or clusters in many colleges, including IITs and IISc | Ask your department or a professor's lab. You usually need a faculty sponsor and must share fairly. |
| Research credit programmes | Some cloud providers and AI companies offer credits or free access for academic research | They open and close over time. Search for current ones and read the conditions. A clear one-page proposal helps. |
| Collaborations | A lab or company partner runs the large experiments | Agree early on data use, authorship and what you can publish. |
Efficient code: get more from the same GPU
- Measure first: time one training step. Find out if the GPU is busy or waiting for data.
- Prepare data once: tokenise or preprocess once and save it, instead of redoing it every run.
- Mixed precision: most modern frameworks support lower-precision training, which often uses less memory and runs faster. Check your framework's docs.
- Gradient accumulation: if a big batch does not fit, add up gradients over several small batches.
- Checkpoints: save regularly so a stopped session does not lose hours of work.
- Reuse results: cache model outputs you evaluate many times, and log every run so you never rerun something by accident.
Estimate before you run
Measured: 1 training run on the small setting = about 40 minutes
Runs needed:
methods (baseline, mine) 2
ablations (no A, no B) + 2
= 4 settings
seeds per setting x 3
= 12 runs
Total: 12 x 40 min = 8 hours of GPU time
Add about 50% for debugging and reruns = about 12 hours
Fits a free tier over a few days? Maybe, with checkpoints.
Plan B: fewer ablations first, add them only if the main result holds.When do you really need more compute?
- The pipeline still has bugs
- You haven't tried a smaller model
- Each run is slow because data loading is slow
- You are unsure what the hypothesis is
- The effect only appears above a certain model size, and you have evidence for that
- Small runs work and the next question is about scale
- You need enough seeds for a trustworthy claim
- A reviewer or mentor asks for a larger check of a clear result
Watch and read more
Lab
A written compute plan for one experiment, with a measured or estimated run time, total hours and a plan B.
The problem
Pick an experiment you want to run: your reproduction target, or an idea of your own. You can do this on paper.
- List every setting you need: methods, baselines, ablations.
- Choose the number of seeds per setting and say why.
- Time one short run if you can (even on a laptop with a tiny subset). If not, write an honest guess and mark it as a guess.
- Multiply to get total GPU hours, then add a margin for debugging.
- Say where you will run it (laptop, free tier, college server) and what could stop you.
- Write a plan B that fits in half the budget: which runs you would drop first, and why.
You are done when
Hard questions
Try to answer before you reveal. If you can answer these, you understood the lesson.
Q1To save compute, a friend suggests running the baseline with 1 seed and your method with 5. Is that fair?Reveal
No. The comparison is then uneven: you know your method's spread but not the baseline's, and one baseline run could be lucky or unlucky. Use the same number of seeds, and the same seeds, for every method you compare. Cut ablations or settings before you cut fairness.
Q2An idea works on the two smallest model sizes you can afford. Can you claim it works for large models?Reveal
Not directly. You can say it held at the sizes you tested, and show the trend. Some effects shrink, grow or reverse with scale. State the sizes clearly in your limitations, and invite others with more compute to test larger models.
Questions people ask
Can I do AI research with only a laptop?
Yes, for many questions. You can run small models, study evaluation, analyse datasets, and do interpretability work on small networks. Use free or shared GPUs for the parts that need them. Start with questions that fit what you have.
Which free GPU service should I use?
Offers change often, so we don't name one. Look at what free notebook services and your college currently offer, compare their limits, and read the terms. Save checkpoints often, because free sessions can end without warning.
Are results on small models taken seriously?
Often yes, if the question fits the scale and the experiment is clean. Be honest in your limitations that larger models may behave differently. A clean small result is more useful than a messy large one.
How do I apply for research credits?
Search for current research credit or academic access programmes from cloud providers and AI companies. Most ask for a short proposal: your question, why it matters, what compute you need and why. A faculty supporter helps. Read the terms on data use and publication.
Lesson test
5 questions. Get 3 right (60%) to pass and complete this lesson.