About Me

I am a third year Ph.D. student in the College of Engineering and Computer Science (CECS) at the Australian National University, where I am supervised by Prof. Liang Zheng and Prof. Stephen Gould. I am currently an intern in the Imagine-Core team at Meta GenAI under Prof. Michael Cohen, working on subject-consistent multiple-shot video generation.

Ph.D. Research Interests: Controllable image synthesis and editing, Creative content generation, Subject-consistent visual storytelling, Long-form video generation.

Previously I was also a two-time intern at Adobe Research working under Dr. Zhe Lin and Dr. Jianming Zhang. Prior to that, I had graduated from Masters of Machine learning and Computer Vision with top-honours from Australian National University. Even earlier, I was worked as machine learning research engineer at Yahoo Japan. I received my Bachelors in Electrical Engineering with specialization in intelligent and cognitive systems from IIT Delhi.

News

[April, 2025] Released R2E-Gym 🚀 - Achieved 51% success rate on SWE-Bench, setting new SOTA for open-weight SWE agents!
[April, 2025] Released REPA-E 🎨 - Enabling stable end-to-end training of latent diffusion transformers with SOTA generation performance and 45x accelerated training!
[March, 2025] 1 paper accepted to CVPR 2025 🎉
[Jan, 2025] Three papers (1 Oral and 2 Poster) accepted at ICLR 2025 🎉🎉🎉
[May, 2024] Started Internship at Meta GenAI working on subject-consistent video generation 🤗

Selected Publications

	R2E-Gym: Procedural Environment Generation and Hybrid Verifiers for Scaling Open-Weights SWE Agents Naman Jain, Jaskirat Singh,** Manish Shetty, Liang Zheng, Koushik Sen, Ion Stoica *Equal Contribution ¹UC Berkeley, ²Australian National University 🏆 Best open-weight SWE agent with 51% success rate — 19% higher than previous open-weight models! [Paper] [Code 🤗] [Project Website] [Data & Models]
	REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers Xingjian Leng, Jaskirat Singh,** Yunzhong Hou, Zhenchang Xing, Saining Xie, Liang Zheng *Project Lead (Equal Contribution) ¹Australian National University, ²Data61-CSIRO, ³New York University 🏆 Enabling stable end-to-end training of latent diffusion transformers with SOTA generation performance and 45x accelerated training! [Paper] [Code] [Project Website] [🤗 Models]
	Negative Token Merging: Image-based Adversarial Feature Guidance Jaskirat Singh, Lindsey Li, Weijia Shi, Ranjay Krishna, Yejin Choi, Pang Wei Koh, Michael F. Cohen, Stephen Gould, Liang Zheng, Luke Zettlemoyer. ¹University of Washington, ²Allen Institute for AI, ³Australian National University 🏆 Image based classifier-free guidance [instead of text] to improve output diversity, quality, control in just few lines of code! [Paper] [Code] [Project Website] [Huggingface Demo 🤗]
	OpenDevin: An Open Platform for AI Software Developers as Generalist Agents Best LLM Agent [🥇 SWE-Bench-Lite] [🥇 SWE-Bench-Verified] ICLR 2025: Oral [Paper] [Code] [Project Website]
	Storybooth: Training-free Multi-Subject Consistency for Improved Visual Storytelling Jaskirat Singh, Junshen Kevin Chen, Jonas Kohler, Michael Cohen ¹Meta GenAI, ²Australian National University ICLR 2025 [Paper]
	Vec2Face: Scaling Face Dataset Generation with Loosely Constrained Vectors Haiyu Wu, Jaskirat Singh, Sicong Tian, Liang Zheng, Kevin Bowyer. ICLR 2025 [Paper] [Code] [Project Website]
	SmartMask: Context Aware High-Fidelity Mask Generation for Fine-grained Object Insertion and Layout Control Jaskirat Singh, Jianming Zhang, Qing Liu, Cameron Smith, Zhe Lin, Liang Zheng ¹Adobe Research, ²Australian National University CVPR 2024, US Patent [Paper] [Project Website]
	Divide, Evaluate, and Refine: Evaluating and Improving Text-to-Image Alignment with Iterative VQA Feedback Jaskirat Singh and Liang Zheng. NeurIPS 2023 [Paper] [Code] [Project Website]
	IMPUS: Image Morphing with Perceptually-Uniform Sampling Using Diffusion Models ICLR 2024 [Paper] [Code]
	High-Fidelity Guided Image Synthesis with Latent Diffusion Models Jaskirat Singh, Stephen Gould, and Liang Zheng. CVPR 2023 [Paper] [Code] [Project Website]
	Paint2Pix: Interactive Painting based Progressive Image Synthesis and Editing Jaskirat Singh, Liang Zheng, Cameron Smith, Jose Echevarria ¹Adobe Research, ²Australian National University ECCV 2022 [Paper] [Code] [Project Website]
	Intelli-Paint: Towards Developing More Human-Intelligible Painting Agents Jaskirat Singh, Cameron Smith, Jose Echevarria, Liang Zheng ¹Adobe Research, ²Australian National University ECCV 2022, US Patent [Paper] [Project Website]
	Combining Semantic Guidance and Deep Reinforcement Learning for Generating Human Level Paintings Jaskirat Singh and Liang Zheng. CVPR 2021 [Paper] [Code]
	Enhanced Scene Specificity with Sparse Dynamic Value Estimation Jaskirat Singh and Liang Zheng arXiv [Paper]
	Dynamic Value Estimation for Single-Task Multi-Scene Reinforcement Learning Jaskirat Singh and Liang Zheng arXiv [Paper]