Ayush Kaushal

Ayush Kaushal

I study sample- and compute-efficient learning in systems that improve as they condition on more context: meta-training of in-context learners, continual pretraining, and scaling analysis that characterizes their behaviour.

I'm a PhD student at Mila and Université de Montréal, advised by Irina Rish. Previously, I co-founded Nolano, where my work focused on low-bit pretraining and inference for foundation models, including the Spectra suite of ternary language models (ICLR 2025 Spotlight) and TriTera (ACL 2025), which added scaling laws, training on 1.2T tokens, and fast CPU and GPU inference for ternary models. I hold an MS from UT Austin and a B.Tech from IIT Kharagpur, and was a research intern at Google, where I worked on encoder architectures for streaming sequence tagging.

Research Timeline

  1. 2025–now

    Mila and Université de Montréal

    Showed that most of the cost of pretraining tabular foundation models comes from training practices inherited from other modalities, and built MegaTabICL: a pretraining framework and stabilized BF16 recipe that match TabICLv2's training trajectory at 3.86× lower stage-1 cost. With it, ran, to our knowledge, the first billion-parameter compute-optimal scaling study of tabular foundation models, which shows the compute-optimal model size growing with the training budget.

  2. 2023–2026

    Nolano

    • 4-bit LLaMA inference on a laptop CPU, implemented on top of ggml and shown publicly on March 10, 2023, two weeks after LLaMA's release.
    • LoRD: one-shot low-rank decomposition that compresses StarCoder 16B to 13.2B parameters with HumanEval pass@1 within 0.1 points of the original.
    • Hi-NOLIN: a 9B English-Hindi model trained on 3,072 GPUs of the Summit supercomputer. Continued pretraining added Hindi while improving English and code, and the model outperforms the larger Pythia 12B.
    • Spectra: the first open suite of language models spanning multiple bit widths, showing that a 3.9B ternary model matches its FP16 counterpart across benchmarks while using fewer bits than an 830M FP16 model.
    • TriTera (Spectra 1.1): scaling laws for ternary language models, ternary models trained on 1.2T tokens, lossless 1.6-bit and 2-bit weight packings now in llama.cpp, and GPU kernels that approach the 8× memory-bound speedup.
  3. 2022

    Google

    Proposed HEAR, a hybrid unidirectional-bidirectional encoder with adaptive restarts for streaming sequence tagging. It reduces streaming compute by 58.9% on average across four tasks (up to 71.1%) while keeping offline accuracy and improving streaming exact match by up to 10 points.

  4. 2021–2023

    UT Austin

    Probed what subword embeddings encode about their constituent characters, and traced how pretraining induces that information.

  5. 2017–2021

    IIT Kharagpur

    Stance detection (NAACL 2021) and the causal direction of data collection in NLP (EMNLP 2021); contributed to JuliaText through Google Summer of Code.

Short CV

News

Recent publications

A complete list is on Google Scholar.

* Equal contribution.

Blog