Armeet profile sketch

Armeet Singh Jatyani


Research Fellow
@Anima AI + Science Lab
[first name] [at] caltech [dot] edu


Computer Science undergrad at Caltech researching AI, computer vision, neural operators, and medical imaging. I recently presented my work at CVPR 2025 and NeurIPS 2024, where I had the chance to meet some pretty remarkable people!


At the Anima AI + Science Lab, I'm working on projects at the intersection of AI and science! Specifically, I'm focusing on developing new fundamental kernel architectures for 3D MRI reconstruction and increasing the potential clinical impacts of my work.


I developed new neural operator architectures for medical imaging, as co-first-author of the U-shaped DISCO Neural Operator (UDNO) paper, an in-place resolution agnostic replacement for the U-Net. Last year, I worked on an RL project, training asymmetric soft-actor critic models for high-speed autonomous racing. We utilized DINO foundation models to extract high-level semantic features like depth and object segmentation to develop a vision-only policy (no need for on-board GPS or LiDAR). The year before that, I imaged hidden fault lines under the nearby Long Beach at the Caltech Seismo Lab, writing a robust data processing pipeline to process 170 TB of seismic data from a large live seismic array of 5,000 sensors. I also worked on models for chemical drug analysis for phramaceutical applications, detecting molecular composition of drugs from Raman spectra at the Wei Bio-Imaging Lab.