Machine Learning Researcher

Nilotpal Kakati

Weizmann Institute · ATLAS Collaboration, CERN Building toward world models for physical and robotic systems, on the back of machine-learning research for the ATLAS experiment.
World Models Representation Learning Generative Models Foundation Models
Nilotpal Kakati

From particle physics
to world models.

I'm a machine-learning researcher with a PhD from the ATLAS experiment at CERN, where several methods I led are in production — including the primary jet flavour tagger for Run 3 (Nature Communications, 2026) and a CUDA graph-construction pipeline delivering orders-of-magnitude speedup over CPU baselines.

My current focus is extending that work toward learned world models for physical and robotic systems. In the last year of my PhD, I worked with Robert Katzschmann's group at ETH Zürich on the design of a flow-matching latent-dynamics pipeline for soft-robotic control; two fellowship proposals came out of that effort (MSCA top-quartile; ETH AI top-10%). The architectural patterns from particle-physics ML map directly to contact dynamics — that bridge is the foundation for what I want to do next.

Now

Postdoctoral Researcher

Weizmann Institute of Science. Extending physics-guided ML toward learned world models.

2025

PhD — Particle Physics

Defended at the Weizmann Institute of Science. ATLAS Collaboration, CERN.

2024

ATLAS Outstanding Achievement Award

For contributions to jet flavor tagging with graph neural networks.

2023–24

FGS Dean Award of Excellence

Weizmann Institute of Science — for excellence in physics, awarded to international doctoral students.

2020

Joined ATLAS at CERN

Began PhD research on machine learning for collider physics.

Research & Engineering.

Selected publications.

  1. 01
    Transforming jet flavour tagging at ATLAS ATLAS Collaboration · Nat. Commun. 17, 541 (2026)
  2. 02
    HGPflow: Extending Hypergraph Particle Flow to Collider Event Reconstruction Kakati et al. · Eur. Phys. J. C 85, 847 (2025)
  3. 03
    Advancing set-conditional set generation: Diffusion models for fast simulation of reconstructed particles Kobylianskii, Soybelman, Kakati et al. · Phys. Rev. D 110, 092013 (2024)
  4. 04
    TIGER: Topology-Independent Graph-based Event Reconstruction Soybelman, Di Bello, Kakati et al. · Phys. Rev. D 113, 012014 (2026)
  5. 05
    Automatizing the search for mass resonances using BumpNet Kakati et al. · JHEP 02 (2025) 122
  6. 06
    Measurements of WH and ZH production with H→bb̄ and direct constraints on the charm Yukawa coupling ATLAS Collaboration · JHEP 04 (2025) 075

Selected talks.

  1. 01
    To Solve Particle Physics with One Model Build Big or Build Smart, MIAPbP, Munich · 2025
  2. 02
    ML for Combined Performance Studies in ATLAS on behalf of ATLAS · LHCP, Taipei · 2025
  3. 03
    Flavor Tagging with ATLAS and CMS at the HL-LHC on behalf of ATLAS & CMS · Higgs 2022, Pisa
  4. 04
    Denoising Graph Super-Resolution with Diffusion Models and Transformers ML4Jets, Paris · 2024
  5. 05
    Jet Flavor Tagging Using Graph Neural Networks on behalf of ATLAS · Connecting the Dots, Princeton · 2022

Let's connect.

Looking for my next role in fundamental ML — world models, learned simulators, foundation models for physical systems.