Kia Rahmani

Kia Rahmani

Member of Technical Staff · Math, Inc. · New York, NY

I work at Math, Inc., at the intersection of formal verification and AI.

Before that I spent a few years at a startup building neurosymbolic program synthesizers and LLM agents. I did my Ph.D. at Purdue and a postdoc at UT Austin, both in programming languages and formal methods, and spent time at Microsoft Research.

Email Google Scholar GitHub LinkedIn

Publications

  1. 2024

    Dynamic Model Predictive Shielding for Provably Safe Reinforcement Learning

    A. Banerjee, K. Rahmani, J. Biswas, I. Dillig

    NeurIPS ·Conference on Neural Information Processing Systems · arXiv

  2. 2024

    Programming-by-Demonstration for Long-Horizon Robot Tasks

    N. Patton, K. Rahmani, M. Missula, J. Biswas, I. Dillig

    POPL ·Symposium on Principles of Programming Languages · ACM · pdf

  3. 2024

    Programmatic Imitation Learning from Unlabeled and Noisy Demonstrations

    J. Xin, L. Zheng, K. Rahmani, J. Wei, J. Holtz, I. Dillig, J. Biswas

    IEEE RA-L ·IEEE Robotics and Automation Letters · IEEE · project · video · arXiv

  4. 2023

    Multi-modal Program Inference

    K. Rahmani, M. Raza, S. Gulwani, V. Le, D. Morris, A. Radhakrishna, G. Soares, A. Tiwari

    US Patent US20230176829A1 Patent · patent

  5. 2022

    Symbolic Analysis of Weak Concurrency Semantics in Modern Database Programs

    K. Rahmani

    Ph.D. thesis, Purdue University Thesis · thesis · slides

  6. 2021

    Multi-modal Program Inference: a Marriage of Large Language Models and Component-based Synthesis

    K. Rahmani, M. Raza, S. Gulwani, V. Le, D. Morris, A. Radhakrishna, G. Soares, A. Tiwari

    OOPSLA ·Object-Oriented Programming, Systems, Languages, and Applications · ACM · arXiv · pdf

  7. 2021

    Repairing Serializability Bugs in Distributed Database Programs via Automated Schema Refactoring

    K. Rahmani, K. Nagar, B. Delaware, S. Jagannathan

    PLDI ·Conference on Programming Language Design and Implementation · ACM

  8. 2019

    CLOTHO: Directed Test Generation for Weakly Consistent Database Systems

    K. Rahmani, K. Nagar, B. Delaware, S. Jagannathan

    OOPSLA ·Object-Oriented Programming, Systems, Languages, and Applications · arXiv

Service

  1. 2026

    AI for Verifiable Coding

    VeriCodeGen @ NeurIPS · Reviewer

  2. 2026

    ACM Conference on Object-Oriented Programming, Systems, Languages, and Applications

    OOPSLA · Review committee

  3. 2025

    International Conference on Learning Representations

    ICLR · Program committee

  4. 2025

    International Conference on Machine Learning

    ICML · Program committee

  5. 2024

    Conference on Neural Information Processing Systems

    NeurIPS · Program committee

  6. 2024

    IEEE Robotics and Automation Letters

    IEEE RA-L · Reviewer

  7. 2024

    U.S. Department of Energy SBIR/STTR program

    DoE SBIR/STTR · Reviewer

  8. 2023

    International Symposium on Technological Advances in Human-Robot Interaction

    TAHRI · Program committee

  9. 2023

    IEEE/RSJ International Conference on Intelligent Robots and Systems

    IROS · Program committee

  10. 2021

    Advances in Programming Languages and Neurosymbolic Systems

    AIPLANS @ NeurIPS · Program committee