This is the website of Alif Munim. I am a machine learning specialist at the University Health Network, building AI foundation models for cardiology & clinical trial matching. I am best known for my work on EchoJEPA; MedRAX; and AI safety & alignment.

Previously, I worked on IBM Watson; interned at the Vector Institute; and conducted research at the University of Toronto & Toronto Metropolitan University. I graduated with distinction with a bachelor of science (BSc) degree in computer science in 2023, with a minor in psychology. I also serve as Community Lead for AI Safety & Alignment at Cohere Labs.

News

Papers

Google Scholar

2026 EchoJEPA: A Latent Predictive Foundation Model for Echocardiography A Munim, A Fallahpour, T Szasz, A Attarpour, R Jiang, B Sooriyakanthan, M Sooriyakanthan, H Whitney, J Slivnick, Q Garrido, K Sinha, W Tsang, B Rubin, B Wang arXiv preprint arXiv:2602.02603 · arXiv · website
2025 MedRAX: Medical Reasoning Agent for Chest X-ray A Fallahpour, J Ma, A Munim, H Lyu, B Wang arXiv preprint arXiv:2502.02673 · arXiv
2025 Benchmarking and Adapting On-Device Large Language Models for Clinical Decision Support A Munim, J Ma, O Ibrahim, A Abdalla, S Yin, L Chen, B Wang arXiv preprint arXiv:2601.03266 · arXiv
2023 Interpretable Machine Learning for Automated Left Ventricular Scar Quantification in Hypertrophic Cardiomyopathy Patients Z Navidi, J Sun, RH Chan, K Hanneman, A Al-Arnawoot, A Munim, et al. PLOS Digital Health 2(1), e0000159
2023 Bringing the State-of-the-Art to Customers: A Neural Agent Assistant Framework for Customer Service Support S Obadinma, FK Khattak, S Wang, T Sidhom, E Lau, S Robertson, J Niu, ... EMNLP 2022 (Industry Track)

Projects

Research

Community & Industry

EchoJEPA: A Latent Predictive Foundation Model for Echocardiography

With Adib Fallahpour, Ahmadreza Attarpour, Teodora Szasz, River Jiang, et al. · UHN · Supervised by Dr. Bo Wang

Built on Yann LeCun's JEPA architecture, EchoJEPA predicts the meaning of masked cardiac video regions in an abstract latent space rather than reconstructing pixels. Trained on 18 million ultrasound videos from 300K patients at UHN. 20% lower error on cardiac function estimation; 79% view classification with 1% of labels; zero-shot pediatric transfer beats all baselines.

Foundation Models Echocardiography JEPA Self-Supervised

MedRAX: Medical Reasoning Agent for Chest X-ray

With Adib Fallahpour, Jingyi Ma, Huizi Lyu · UHN · Supervised by Dr. Bo Wang

The first AI agent framework that integrates specialized chest X-ray analysis tools with large language models via a ReAct reasoning loop. Coordinates CheXagent, MedSAM, and other clinical tools to perform complex multi-step medical reasoning. Achieves SOTA on ChestAgentBench (63.1%) and SLAKE VQA (90.35%) without additional training.

AI Agents Medical Reasoning Chest X-ray Tool Use

Benchmarking On-Device LLMs for Clinical Decision Support

With Jingyi Ma, Omar Ibrahim, Aya Abdalla, Shuo Yin, Ling Chen · UHN · Supervised by Dr. Bo Wang

Evaluating whether on-device LLMs (20B and 120B parameters) can match cloud-based systems for clinical tasks while preserving patient privacy. Designed three evaluation frameworks: generalist medical reasoning, specialist clinical scoring, and LLM-as-judge. Fine-tuning the 20B model boosts accuracy from 77% to 87%, approaching GPT-5 and beating DeepSeek-R1.

On-Device LLMs Clinical AI Privacy Fine-Tuning

Towards Generative Models for Medical Imaging

Undergraduate Thesis · Toronto Metropolitan University · 2023

An investigation into diffusion models for generating realistic medical images to address data limitations in healthcare. Experiments showed that training from scratch on domain-specific data consistently outperformed fine-tuning pretrained natural-image models. Reviews GANs, VAEs, and diffusion model foundations.

Diffusion Models Medical Imaging Generative AI

Classical vs Deep Learning for Brain MRI Segmentation

With Dr. Dafna Sussman, Daniel Nussey, Rachita Singh · Toronto Metropolitan University

Compared U-Nets, fully convolutional networks, and gradient-boosted ensemble models for brain metastasis segmentation in <100 3D MRI scans. Deep learning significantly outperformed classical methods; data augmentation proved critical for generalization.

Medical Imaging Segmentation U-Net

Writing & Links

EchoJEPA: A Latent Predictive Foundation Model for Echocardiography 2026
MedRAX: Medical Reasoning Agent for Chest X-ray 2025
On-Device LLMs for Clinical Decision Support 2025
AI Safety & Alignment at Cohere Labs 2024
Towards Generative Models for Medical Imaging 2023
Classical vs Deep Learning for Brain MRI Segmentation 2022
Email Speech Act Classification for Task Automation 2021