Aviraj Newatia

Aviraj Newatia

PhD student in Computer Science
University of Cambridge · CAMRAL

Hi, I’m Avi. I’m a PhD student in Computer Science at the University of Cambridge, in the Cambridge Resilient Autonomous Learning Group (CAMRAL), supervised by Dr Rika Antonova. My PhD is funded by ARIA’s Robot Dexterity programme as part of the Ada-GO project.

I work on simulators, physics-informed learning and robotics: how to adapt simulators, solvers and learned policies without breaking the dynamics they model. Before this I worked on machine unlearning with Dr Rahul Krishnan at the Vector Institute, and on quantum circuit design with Dr Hans-Arno Jacobsen’s group. I did my BSc in Computer Science and Mathematics at the University of Toronto.

Outside research you will usually find me running, swimming, reading, or writing. Occasionally I also like to go climbing, play tennis, squash, or some padel.

EmailCVGoogle ScholarGitHubORCIDSubstack

News

  1. Oct 2026 · PaperStructural Affinity, an algorithm for coherent cage-based shape design, is on arXiv, with code.
  2. Oct 2026 · PaperCo-design Gym, a benchmark for embodiment–policy co-optimisation, is on arXiv.
  3. Apr 2026 · PaperForget set-free unlearning (Reload) published at ICLR 2026.
  4. Mar 2026 · AwardReceived an NVIDIA Academic Grant for simulation and modelling research.
  5. Feb 2026 · PaperProbabilistic design of parametrised quantum circuits is on arXiv.
Show older (6)
  1. Oct 2025 · MilestoneStarted my PhD at Cambridge with Rika Antonova, funded by ARIA's Robot Dexterity programme.
  2. Jun 2025 · New postRaft, Paper, Scissors — A foray into distributed systems through eventuaaal convergence.
  3. Jun 2025 · MilestoneGraduated from the University of Toronto (BSc, Computer Science and Mathematics).
  4. Dec 2024 · PaperTabular unlearning without a forget set appeared at the TRL workshop, NeurIPS 2024.
  5. Oct 2024 · New postSelf-learning Machines based on Hamiltonian Echo Backpropagation — A read of this paper on physical self-learning machines and their theorization
  6. Aug 2024 · TalkDemoed Reload, selective data forgetting and replacement, at Vector Institute Demo Day.

Selected publications

Image

Structural Affinity: An Algorithm for Coherent Cage-Based Shape Design

A. Newatia, S. Yang, H. Zhong, R. Antonova

arXiv preprint

arXivProjectCode
Image

Co-design Gym: A Unified Benchmark for Embodiment-Policy Co-optimization

A. Newatia*, Y. Tsvetkov*, L. Pleiss, A. Spielberg, R. Antonova

arXiv preprint* equal contribution

arXivProject
1 · trained weights

Mitigating Privacy Risk via Forget Set-Free Unlearning

A. Newatia, M. Cooper, V. Nguyen, R. G. Krishnan

ICLR 2026

arXivCodePoster
trained on the full table

Unlearning Tabular Data Without a "Forget Set"

A. Newatia, M. Cooper, R. G. Krishnan

Table Representation Learning Workshop, NeurIPS 2024

Paper
All publications →

Latest writing

All writing →

Activity

Background

Experience 8
  • 2025 – nowPhD researcherCAMRAL, University of CambridgeCage-based deformation for computational shape design, and Co-design Gym, in the ARIA-funded Ada-GO project.
  • 2024 – 2026Research intern, then collaboratorVector Institute, with Rahul KrishnanForget set-free unlearning with gradient-based weight saliency and sparsity; led to the NeurIPS 2024 workshop and ICLR 2026 papers.
  • 2024 – 2026Computer systems researcherMiddleware Systems Research Group, University of TorontoEvolutionary architecture search for quantum ML circuits, and the effect of error mitigation on QML models.
  • 2024Bioinformatics researcherRNA Lab, University of TorontoText embeddings and a contrastive method to cluster Sequence Read Archive metadata.
  • 2022 – 2024Machine learning researcherBMO Lab, University of TorontoFine-tuned Llama-2 on play scripts; built a real-time VQGAN+CLIP canvas with Whisper voice control.
  • 2022Data catalogue research assistantSuperCDMS, SNOLABJMeter test suites that found pipeline bottlenecks; fixes cut request time from 18 s to 0.8 s.
  • 2022NLP internYellow AI, New DelhiUnsupervised multilingual intent clustering of customer-support queries.
  • 2019 – 2021R&D internNike and Rookie India, RJ CorpAssociation mining on 2M+ transactions and an RNN customer-feedback classifier.
Education 3
  • 2025 – nowPhD, Computer ScienceUniversity of Cambridge, Emmanuel CollegeSupervised by Rika Antonova. Funded by ARIA's Robot Dexterity programme.
  • 2021 – 2025BSc (Hons), Computer Science and MathematicsUniversity of TorontoDean's List all four years. Graduate courses in computational imaging, visual computing systems and deep learning theory.
  • – 2021International Baccalaureate, 42/45The British School, New Delhi
Talks 1
  • Aug 2024Reload: selective data forgetting and selective data replacementVector Institute Demo Day, Toronto
Service & leadership 4
  • 2026 – nowComputer Officer, MCREmmanuel College, Cambridge
  • 2025 – nowRowerEmmanuel College Boat Club
  • 2024 – 2025Director of Academic AffairsComputer Science Student Union, University of TorontoFounded UofT St George's first STEM undergraduate research journal and ran the undergraduate colloquium.
  • 2024 – 2025Undergraduate memberDepartment of Computer Science Undergraduate Affairs Committee, University of Toronto