Ahmed Elhag

prof_pic.jpg

I am a PhD student at the Department of Computer Science at the University of Oxford, advised by Michael Bronstein. My research focuses on the intersection of geometric deep learning, graph ML, and generative models. I’m particularly interested in how we can combine these approaches to develop robust ML methods that can accelerate progress in drug discovery and molecular design. One of my current research areas is learning symmetry and equivariance in unconstrained models to address the limitations of strictly equivariant architectures (see Relaxed Equivariance via Multitask Learning and Learning Inter-Atomic Potentials without Explicit Equivariance).

Before my doctoral studies, I graduated from the African Masters of Machine Intelligence program at AIMS Senegal. I also interned at Apple MLR team where I worked on developing generative models for 3D and graph-structured data.

Contact: ahmed.elhag[at]cs.ox.ac.uk

Selected Publications

  1. Learning Inter-Atomic Potentials without Explicit Equivariance
    Ahmed A. Elhag*, Arun Raja*, Alex Morehead*, and 3 more authors
    2025
  2. LOG
    Relaxed Equivariance via Multitask Learning
    Ahmed A. Elhag, T. Konstantin Rusch, Francesco Di Giovanni, and 1 more author
    In LOG, 2025
  3. ICLR
    Manifold Diffusion Fields
    Ahmed A. Elhag, Yuyang Wang, Joshua M. Susskind, and 1 more author
    In ICLR, 2024
  4. ICML
    Swallowing the Bitter Pill: Simplified Scalable Conformer Generation
    Yuyang Wang, Ahmed A. Elhag, Navdeep Jaitly, and 2 more authors
    In ICML, 2024