Juan Camilo Ramirez De Los Rios

About

I am a third year PhD candidate at Mila and DIRO under the supervision of Simon Lacoste-Julien.

I develop constrained learning algorithms for training neural networks, which enable enforcing requirements such as fairness, sparsity and safety in LLMs.

My research aims to ensure that training models with constraints is as easy as training without them.

Check Cooper out, our library for non-convex constrained optimization in PyTorch.

Previously, I was an intern in Simon's group at Mila under the supervision of Jose Gallego-Posada. Before that, I completed a BSc in Mathematical Engineer at Universidad EAFIT. During the BSc, I spent a summer at McKinsey & Co. as a research intern.

Research interests: Non-convex Constrained Optimization, Min-Max Optimization, Machine Unlearning, Sparsity, Fairness.

Contact: juan43ramirez (at) gmail (dot) com

News

2025

Apr 1: We just released version 1.0.0 of Cooper, a PyTorch library for non-convex constrained optimization, along with a companion paper.
Jan 22: Our latest paper Feasible Learning has been accepted at AISTATS 2025! Feasible Learning is a novel sample-centric learning paradigm where models are trained by solving a feasibility problem that bounds the loss for each training sample.
Jan 1: Elated to server as Associate co-Chair for ICML 2025.

2024

Nov 1: I have been awarded the Excellence Scholarship by the Association des Diplômés de l'Université de Montréal.
Sep 6: I succeded in my predoctoral exam. I am now a PhD candidate!
Sep 1: I will be a TA for Simon Lacoste-Julien's course on Probabilistic Graphical Models for a second time.
May 1: On PI Controllers for Updating Lagrange Multipliers in Constrained Optimization has been accepted at ICML 2024! Vienna, here we come again.
Mar 04: Delighted to receive Université de Montréal's AI scholarship 2023-2024!
Jan 29: Excited to be a TA at the first edition of the Tropical Probabilistic AI School in Rio de Janeiro, Brazil! I am happy to see more AI events popping up around Latin America.
Jan 16: Balancing Act: Constraining Disparate Impact in Sparse Models has been accepted at ICLR 2024! Vienna, here we come.
Jan 12: Quentin Bertrand and I will be TAing Gauthier Gidel's Adversarial Machine Learning course at Mila and the University of Montreal.

Dec 7: Excited to give a talk at Montréal Machine Learning and Optimization (MTL MLOpt) about our recent work Balancing Act: Constraining Disparate Impact in Sparse Models.
Sep 4: Together with António Góis, I will be TAing Simon Lacoste-Julien's graduate course on Probabilistic Graphical Models at Mila.
Jul 23: Arriving in Honolulu, Hawaii, where I will have the pleasure to present Omega: Optimistic EMA Gradients. Omega, our method for stochastic min-max optimization, was awarded an oral at the LatinX in AI workshop at ICML 2023!
Jan 27: I will attend Khipu 2023 in Montevideo, Uruguay. Glad to be a part of the Latin American AI community.

2022

Oct 14: See you at the PyTorch conference 2022 in New Orleans! Cooper will be making an appearance (poster).
Sep 14: I will be presenting my first NeurIPS paper: Controlled Sparsity via Constrained Optimization or: How I Learned to Stop Tuning Penalties and Love Constraints in New Orleans.
Sep 05: Excited to start a PhD in Artificial Intelligence under the supervision of Simon Lacoste-Julien at Mila and the University of Montreal.
Jul 8: Attending the Sparsity in Neural Networks workshop. We are presenting our latest work on implicit neural representations: L₀onie: Compressing COINs with L₀-constraints.

2021

Dec 17: Flexible Learning of Sparse Neural Networks via Constrained L0 Regularization, won the best poster award at the LatinX in AI Workshop @ NeurIPS 2021!
Nov 1: Starting an internship at Mila. I will join Simon Lacoste-Julien's group under the supervision of Jose Gallego-Posada.
Jul 1: I graduated from the BSc in Mathematical Engineering at Universidad EAFIT! Plus, got a mention for my contributions to the student organization. Check out my Linkedin post about it.
Jan 14: I am auditing Ioannis Mitliagkas' graduate course on Deep Learning Theory at Université de Montréal.

2020 and beyond

Jul 2020: Jose Gallego-Posada, PhD student at Mila will be my supervisor for my undergraduate thesis. I will be working on deep generative models.
Jun 2019: Now part of McKinsey & Co. in Belgium as a research intern. Working with Antoine Stevens and Patrick Dehout in ML for the agricultural and chemical industries.
Jan 2019: Arrived in Louvain-la-Neuve, Belgium for an exchange semestrer at Université Catholique de Louvain.
Nov 2017: I have been appointed as president of CIGMA-OE, the Mathematical Engineering chapter of the Student Organization at Universidad EAFIT.
Dec 2015: I was awarded a full scholarship for the Bachelor's degree in Mathematical Engineering at Universidad EAFIT.
Dec 2015: Scored amongst the best 0.1% on the Colombian high school examination ICFES.
Dec 2015: Ranked first in the National Chemistry Olympiads of Universidad de Antioquia.

Publications

Preprints

J. Gallego-Posada*, Juan Ramirez*, M. Hashemizadeh* and S. Lacoste-Julien. Cooper: A Library for Constrained Optimization in Deep Learning. arXiv preprint at arXiv:2504.01212, 2025.

Conference

Juan Ramirez*, I. Hounie*, J. Elenter*, J. Gallego-Posada*, M. Hashemizadeh, A. Ribeiro^ and S. Lacoste-Julien^. Feasible Learning. In AISTATS, 2025.
M. Sohrabi*, Juan Ramirez*, T. H. Zhang, S. Lacoste-Julien and J. Gallego-Posada. On PI Controllers for Updating Lagrange Multipliers in Constrained Optimization. In ICML, 2024.
M. Hashemizadeh*, Juan Ramirez*, R. Sukumaran, G. Farnadi, S. Lacoste-Julien and J. Gallego-Posada. Balancing Act: Constraining Disparate Impact in Sparse Models. In ICLR, 2024.
J. Gallego-Posada, Juan Ramirez, A. Erraqabi, Y. Bengio and S. Lacoste-Julien. Controlled Sparsity via Constrained Optimization or: How I Learned to Stop Tuning Penalties and Love Constraints. In NeurIPS, 2022.

Workshop

Juan Ramirez, R. Sukumaran, Q. Bertrand and G. Gidel. Omega: Optimistic EMA Gradients. LatinX in AI workshop at ICML, 2023.
Juan Ramirez and J. Gallego-Posada. L₀onie: Compressing COINs with L₀-constraints. Sparsity in Neural Networks Workshop, 2022.
J. Gallego-Posada, Juan Ramirez and A. Erraqabi. Flexible Learning of Sparse Neural Networks via Constrained L0 Regularization. LatinX in AI Workshop at NeurIPS, 2021.

Teaching Assistantships

Fall 23 and 24: Probabilistic Graphical Models by Simon Lacoste-Julien .
Winter 24: Adversarial Machine Learning by Gauthier Gidel .
Summer schools: Tropical Probabilistic AI School (2024), Lisbon Machine Learning School (2024).

Service

Associate (Program) co-Chair for ICML 2025.
Reviewer for ICML 2025, AISTATS 2025, and NeurIPS 2024. Reviewer for the LatinX in AI workshops at NeurIPS 2024, ICML 2024, and ICML 2023.
Mila 2025 applicant and Khipu 2025 application reviewer.
Social chair for the LatinX in AI workshop at NeurIPS 2024 and visa chair for the LatinX in AI workshop at ICML 2024.
Cosupervised (with Jose) the undergraduate research projects of Daniel Otero and Isabel Urrego, BSc students in mathematical engineering in Colombia (2023).