Machine Learning Researcher

Nilotpal Kakati

Weizmann Institute · ATLAS Collaboration, CERN Moving from machine learning for the ATLAS experiment to learned world models for physical systems.
World Models Representation Learning Generative Models Foundation Models
Nilotpal Kakati

From particle physics
to world models.

I'm a machine-learning researcher and postdoc at the Weizmann Institute, with a PhD from the ATLAS experiment at CERN. Two things I built there run in production: GN2, the jet flavor tagger used across ATLAS (4000+ physicists), whose implementation I led, and the GPU graph construction for track reconstruction, which I co-developed and which cut the slowest step from about ten minutes per event to 69 ms. I also led a multimodal foundation model for particle physics, and I'm first author on HGPflow and calorimeter super-resolution.

My current work is on self-supervised learning. In two first-author manuscripts, I use image simulators where every generating factor is known to test what JEPA-style encoders keep, and whether label-free training signals notice when they lose it. Representation learning and world models are what I want to work on next.

I also teach: lectures and tutorials for Weizmann's Practical Deep Learning for Science course, across four editions between 2021 and 2025.

Now

Postdoctoral Researcher

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

2026

John F. Kennedy Prize

Weizmann School of Science — for outstanding Ph.D. research.

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
  2. 02
    Precision Tokenization for Multi-Modal Data in High Energy Physics Krupa, Klein, Kakati et al. · arXiv:2610.13876 (2026)
  3. 03
    Transforming jet flavour tagging at ATLAS ATLAS Collaboration · Nat. Commun. 17, 541 (2026)
  4. 04
  5. 05
  6. 06
    Advancing set-conditional set generation: Diffusion models for fast simulation of reconstructed particles Kobylianskii, Soybelman, Kakati et al. · Phys. Rev. D 110, 092013 (2024)
  7. 07
    Self-Supervised Learning Strategies for Jet Physics Rieck et al. · Mach. Learn.: Sci. Technol. 6, 045015 (2025)

Selected talks.

  1. 01
    To Solve Particle Physics with One Model Build Big or Build Smart, MIAPbP, Munich · 2025
  2. 02
  3. 03
    Flavor Tagging with ATLAS and CMS at the HL-LHC on behalf of ATLAS & CMS · Higgs 2022, Pisa
  4. 04
  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.