Brandon Amos
Research Scientist
Facebook AI (FAIR)


I am a research scientist at Facebook AI (FAIR) in NYC and study foundational topics in machine learning and optimization, recently involving reinforcement learning, control, optimal transport, and geometry. My research is on learning systems that understand and interact with our world and focuses on integrating structural information and domain knowledge into these systems to represent non-trivial reasoning operations. A key theme of my work in this space involves the use of optimization as a differentiable building block in larger architectures that are end-to-end learned. I believe that science should be open and reproducible and freely publish my research code to GitHub.


Ph.D. in Computer Science, Carnegie Mellon University (0.00/0.00)

Thesis: Differentiable Optimization-Based Modeling for Machine Learning
Advisor: J. Zico Kolter

2014 - 2019
B.S. in Computer Science, Virginia Tech (3.99/4.00)
2011 - 2014

Previous Positions

Research Assistant, Carnegie Mellon University (with J. Zico Kolter)

2016 - 2019

Research Intern, Intel Labs, Santa Clara (with Vladlen Koltun)


Research Intern, Google DeepMind, London (with Nando de Freitas and Misha Denil)


Research Assistant, Carnegie Mellon University (with Mahadev Satyanarayanan)

2014 - 2016

Research Intern, Adobe Research, San Jose (with David Tompkins)


Research Assistant, Virginia Tech (with Layne Watson and David Easterling)

2013 - 2014

Research Assistant, Virginia Tech (with Jules White and Hamilton Turner)

2012 - 2014

Research Assistant, Virginia Tech (with Binoy Ravindran and Alastair Murray)

2012 - 2014

Software Intern, Snowplow, Remote (Scala development)

2013 - 2014

Software Intern, Qualcomm, San Diego (Python and C++ development)


Software Intern, Phoenix Integration, Virginia (C++, C#, and Java development)


Network Administrator Intern, Sunapsys, Virginia


Honors & Awards

ICLR Outstanding Reviewer
2016 - 2019
NSF Graduate Research Fellowship
2011 - 2014
Nine undergraduate scholarships

Roanoke County Public Schools Engineering, Salem-Roanoke County Chamber of Commerce, Papa John's, Scottish Rite of Freemasonry, VT Intelligence Community Conter for Academic Excellence, VT Pamplin Leader, VT Benjamin F. Bock, VT Gay B. Shober, VT I. Luck Gravett


Representative publications that I am a primary author on are highlighted.
[Google Scholar]


1. Tutorial on amortized optimization for learning to optimize over continuous domains [abs] [code]
Brandon Amos
arXiv 2022
2. Cross-Domain Imitation Learning via Optimal Transport [abs]
Arnaud Fickinger, Samuel Cohen, Stuart Russell, and Brandon Amos
ICLR 2022
3. Semi-Discrete Normalizing Flows through Differentiable Tessellation [abs]
Ricky T. Q. Chen, Brandon Amos, and Maximilian Nickel
arXiv 2022


4. On the model-based stochastic value gradient for continuous reinforcement learning [abs] [code] [slides]
Brandon Amos, Samuel Stanton, Denis Yarats, and Andrew Gordon Wilson
L4DC 2021 (Oral)
5. Riemannian Convex Potential Maps [abs] [code] [slides]
Samuel Cohen*, Brandon Amos*, and Yaron Lipman
ICML 2021
6. CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints [abs] [code]
Anselm Paulus, Michal Rolínek, Vít Musil, Brandon Amos, and Georg Martius
ICML 2021
7. Scalable Online Planning via Reinforcement Learning Fine-Tuning [abs]
Arnaud Fickinger, Hengyuan Hu, Brandon Amos, Stuart Russell, and Noam Brown
NeurIPS 2021
8. Aligning Time Series on Incomparable Spaces [abs] [code] [slides]
Samuel Cohen, Giulia Luise, Alexander Terenin, Brandon Amos, and Marc Peter Deisenroth
9. Learning Neural Event Functions for Ordinary Differential Equations [abs] [code]
Ricky T. Q. Chen, Brandon Amos, and Maximilian Nickel
ICLR 2021
10. Neural Spatio-Temporal Point Processes [abs] [code]
Ricky T. Q. Chen, Brandon Amos, and Maximilian Nickel
ICLR 2021
11. Improving Sample Efficiency in Model-Free Reinforcement Learning from Images [abs] [code]
Denis Yarats, Amy Zhang, Ilya Kostrikov, Brandon Amos, Joelle Pineau, and Rob Fergus
AAAI 2021
12. Neural Fixed-Point Acceleration for Convex Optimization [abs] [code]
Shobha Venkataraman* and Brandon Amos*
ICML AutoML 2021
13. Sliced Multi-Marginal Optimal Transport [abs]
Samuel Cohen, Alexander Terenin, Yannik Pitcan, Brandon Amos, Marc Peter Deisenroth, and K S Sesh Kumar
NeurIPS OTML 2021
14. Input Convex Gradient Networks [abs]
Jack Richter-Powell, Jonathan Lorraine, and Brandon Amos
NeurIPS OTML 2021
15. Imitation Learning from Pixel Observations for Continuous Control [abs]
Samuel Cohen, Brandon Amos, Marc Peter Deisenroth, Mikael Henaff, Eugene Vinitsky, and Denis Yarats
NeurIPS DeepRL 2021
16. MBRL-Lib: A Modular Library for Model-based Reinforcement Learning [abs] [code]
Luis Pineda, Brandon Amos, Amy Zhang, Nathan Lambert, and Roberto Calandra
arXiv 2021


17. The Differentiable Cross-Entropy Method [abs] [code] [slides]
Brandon Amos and Denis Yarats
ICML 2020
18. Objective Mismatch in Model-based Reinforcement Learning [abs]
Nathan Lambert, Brandon Amos, Omry Yadan, and Roberto Calandra
L4DC 2020
19. QNSTOP: Quasi-Newton Algorithm for Stochastic Optimization [abs] [code]
Brandon Amos, David Easterling, Layne T. Watson, William Thacker, Brent Castle, and Michael Trosset
20. Neural Potts Model [abs]
Tom Sercu, Robert Verkuil, Joshua Meier, Brandon Amos, Zeming Lin, Caroline Chen, Jason Liu, Yann LeCun, and Alexander Rives
MLCB 2020
21. Deep Riemannian Manifold Learning [abs]
Aaron Lou, Maximilian Nickel, and Brandon Amos
NeurIPS Geo4dl 2020


22. Differentiable Optimization-Based Modeling for Machine Learning [abs] [code]
Brandon Amos
Ph.D. Thesis 2019
23. Differentiable Convex Optimization Layers [abs] [code]
Akshay Agrawal*, Brandon Amos*, Shane Barratt*, Stephen Boyd*, Steven Diamond*, and J. Zico Kolter*
NeurIPS 2019
24. The Limited Multi-Label Projection Layer [abs] [code]
Brandon Amos, Vladlen Koltun, and J. Zico Kolter
arXiv 2019
25. Generalized Inner Loop Meta-Learning [abs] [code]
Edward Grefenstette, Brandon Amos, Denis Yarats, Phu Mon Htut, Artem Molchanov, Franziska Meier, Douwe Kiela, Kyunghyun Cho, and Soumith Chintala
arXiv 2019


26. Learning Awareness Models [abs]
Brandon Amos, Laurent Dinh, Serkan Cabi, Thomas Rothörl, Sergio Gómez Colmenarejo, Alistair Muldal, Tom Erez, Yuval Tassa, Nando de Freitas, and Misha Denil
ICLR 2018
27. Differentiable MPC for End-to-end Planning and Control [abs] [code]
Brandon Amos, Ivan Dario Jimenez Rodriguez, Jacob Sacks, Byron Boots, and J. Zico Kolter
NeurIPS 2018
28. Depth-Limited Solving for Imperfect-Information Games [abs]
Noam Brown, Tuomas Sandholm, and Brandon Amos
NeurIPS 2018
29. Enabling Live Video Analytics with a Scalable and Privacy-Aware Framework [abs]
Junjue Wang, Brandon Amos, Anupam Das, Padmanabhan Pillai, Norman Sadeh, and Mahadev Satyanarayanan


30. OptNet: Differentiable Optimization as a Layer in Neural Networks [abs] [code] [slides]
Brandon Amos and J. Zico Kolter
ICML 2017
31. Input Convex Neural Networks [abs] [code] [slides]
Brandon Amos, Lei Xu, and J. Zico Kolter
ICML 2017
32. Task-based End-to-end Model Learning [abs] [code]
Priya L. Donti, Brandon Amos, and J. Zico Kolter
NeurIPS 2017
33. Quasi-Newton Stochastic Optimization Algorithm for Parameter Estimation of a Stochastic Model of the Budding Yeast Cell Cycle [abs]
Minghan Chen, Brandon Amos, Layne T. Watson, John Tyson, Yang Cao, Cliff Shaffer, Michael Trosset, Cihan Oguz, and Gisella Kakoti
34. You can teach elephants to dance: agile VM handoff for edge computing [abs]
Kiryong Ha, Yoshihisa Abe, Thomas Eiszler, Zhuo Chen, Wenlu Hu, Brandon Amos, Rohit Upadhyaya, Padmanabhan Pillai, and Mahadev Satyanarayanan
SEC 2017
35. An Empirical Study of Latency in an Emerging Class of Edge Computing Applications for Wearable Cognitive Assistance [abs]
Zhuo Chen, Wenlu Hu, Junjue Wang, Siyan Zhao, Brandon Amos, Guanhang Wu, Kiryong Ha, Khalid Elgazzar, Padmanabhan Pillai, Roberta Klatzky, Daniel Siewiorek, and Mahadev Satyanarayanan
SEC 2017
36. A Scalable and Privacy-Aware IoT Service for Live Video Analytics [abs] [code]
Junjue Wang, Brandon Amos, Anupam Das, Padmanabhan Pillai, Norman Sadeh, and Mahadev Satyanarayanan
ACM MMSys 2017 (Best Paper)


37. OpenFace: A general-purpose face recognition library with mobile applications [abs] [code]
Brandon Amos, Bartosz Ludwiczuk, and Mahadev Satyanarayanan
CMU 2016
38. Collapsed Variational Inference for Sum-Product Networks [abs]
Han Zhao, Tameem Adel, Geoff Gordon, and Brandon Amos
ICML 2016
39. Quantifying the impact of edge computing on mobile applications [abs]
Wenlu Hu, Ying Gao, Kiryong Ha, Junjue Wang, Brandon Amos, Zhuo Chen, Padmanabhan Pillai, and Mahadev Satyanarayanan
40. Privacy mediators: helping IoT cross the chasm [abs]
Nigel Davies, Nina Taft, Mahadev Satyanarayanan, Sarah Clinch, and Brandon Amos
HotMobile 2016

2015 and earlier

41. Edge Analytics in the Internet of Things [abs]
Mahadev Satyanarayanan, Pieter Simoens, Yu Xiao, Padmanabhan Pillai, Zhuo Chen, Kiryong Ha, Wenlu Hu, and Brandon Amos
IEEE Pervasive Computing 2015
42. Bad Parts: Are Our Manufacturing Systems at Risk of Silent Cyberattacks? [abs]
Hamilton Turner, Jules White, Jaime A. Camelio, Christopher Williams, Brandon Amos, and Robert Parker
IEEE Security & Privacy 2015
43. Early Implementation Experience with Wearable Cognitive Assistance Applications [abs]
Zhuo Chen, Lu Jiang, Wenlu Hu, Kiryong Ha, Brandon Amos, Padmanabhan Pillai, Alex Hauptmann, and Mahadev Satyanarayanan
WearSys 2015
44. The Case for Offload Shaping [abs]
Wenlu Hu, Brandon Amos, Zhuo Chen, Kiryong Ha, Wolfgang Richter, Padmanabhan Pillai, Benjamin Gilbert, Jan Harkes, and Mahadev Satyanarayanan
HotMobile 2015
45. Are Cloudlets Necessary? [abs]
Ying Gao, Wenlu Hu, Kiryong Ha, Brandon Amos, Padmanabhan Pillai, and Mahadev Satyanarayanan
CMU 2015
46. Adaptive VM handoff across cloudlets [abs]
Kiryong Ha, Yoshihisa Abe, Zhuo Chen, Wenlu Hu, Brandon Amos, Padmanabhan Pillai, and Mahadev Satyanarayanan
CMU 2015
47. Global Parameter Estimation for a Eukaryotic Cell Cycle Model in Systems Biology [abs]
Tricity Andrew, Brandon Amos, David Easterling, Cihan Oguz, William Baumann, John Tyson, and Layne T. Watson
SummerSim 2014
48. Applying machine learning classifiers to dynamic Android malware detection at scale [abs] [code]
Brandon Amos, Hamilton Turner, and Jules White
IWCMC 2013


facebookresearch/amortized-optimization-tutorial | 100 | Tutorial on amortized optimization 2022
facebookresearch/theseus | 210 | Differentiable non-linear optimization library 2022
facebookresearch/mbrl-lib | 567 | Model-based reinforcement learning library 2021
facebookresearch/dcem | 95 | The Differentiable Cross-Entropy Method 2020
facebookresearch/higher | 1.3k | PyTorch higher-order gradient and optimization library 2019
bamos/thesis | 262 | Ph.D. Thesis LaTeX source code 2019
cvxgrp/cvxpylayers | 1.2k | Differentiable Convex Optimization Layers 2019
locuslab/mpc.pytorch | 521 | Differentiable Model-Predictive Control 2018
locuslab/icnn | 236 | Input Convex Neural Networks 2017
locuslab/optnet | 381 | OptNet experiments 2017
locuslab/qpth | 517 | Differentiable PyTorch QP solver 2017
bamos/densenet.pytorch | 736 | PyTorch DenseNet implementation 2017
bamos/block | 261 | Intelligent block matrix constructions 2017
bamos/setGPU | 98 | Automatically use the least-loaded GPU 2017
bamos/dcgan-completion.tensorflow | 1.3k | Image completion with GANs 2016
cmusatyalab/openface | 14.3k | Face recognition with deep neural networks 2015
bamos/zsh-history-analysis | 172 | Analyze and plot your zsh history 2014
cparse/cparse | 242 | C++ expression parser using Dijkstra's shunting-yard algorithm 2013
bamos/cv | 342 | Source for this CV: Creates LaTeX/Markdown from YAML/BibTeX 2013
bamos/dotfiles | 234 | Linux, mutt, xmonad, vim, emacs, zsh 2012

Invited Talks

Columbia University 2021
IBM Research 2021
Max Planck Institute for Intelligent Systems (Tübingen) Seminar 2020
Montreal Institute for Learning Algorithms Seminar 2020
ECCV Deep Declarative Networks Tutorial 2020
CVPR Deep Declarative Networks Workshop 2020
Caltech CS 159, Guest Lecture 2020
SIAM MDS Minisymposium on Learning Parameterized Energy Minimization Models 2020
New York University CILVR Seminar 2019
INFORMS Session on Prediction and Optimization 2019
Facebook AI Research 2019
ISMP Session on Machine Learning and Optimization 2018
Google Brain 2018
Bosch Center for AI 2018
Waymo Research 2018
Tesla AI 2018
NVIDIA Robotics 2018
Salesforce Research 2018
OpenAI 2018
UC Berkeley 2018

Interns and Students

Aaron Lou (visiting FAIR from Cornell and Stanford) 2020 - 2022
Eugene Vinitsky (visiting FAIR from Berkeley) 2021 - 2022
Arnaud Fickinger (visiting FAIR from Berkeley) 2021 - 2022
Samuel Cohen (visiting FAIR from UCL) 2021 - 2022
Ricky Chen (visiting FAIR from Toronto, now: scientist at FAIR) 2020
Paul Liang (visiting FAIR from CMU) 2020
Phillip Wang (at CMU, now: CEO at Gather) 2018

Professional Activities

NeurIPS Learning Meets Combinatorial Optimization Workshop Organizer 2020
CVPR Deep Declarative Workshop Organizer 2020
ECCV Deep Declarative Tutorial Organizer 2020
CMU CSD MS Admissions 2014 - 2015


Neural Information Processing Systems (NeurIPS)
International Conference on Machine Learning (ICML)
International Conference on Learning Representations (ICLR)
IEEE Conference on Computer Vision and Pattern Recognition (CVPR)
IEEE International Conference on Computer Vision (ICCV)
IEEE International Conference on Robotics and Automation (ICRA)
AAAI Conference on Artificial Intelligence (AAAI)
Optimization Letters


Graduate AI (CMU 15-780), TA S2017
Distributed Systems (CMU 15-440/640), TA S2016
Software Design and Data Structures (VT CS2114), TA S2013


Programming C, C++, Fortran, Haskell, Java, Lua, Make, Mathematica, Python, R, Scala
Frameworks JAX, NumPy, Pandas, PyTorch, SciPy, TensorFlow, Torch7
Tools Linux, emacs, vim, evil, org, mu4e, xmonad, git, tmux, zsh

Last updated on 2022-04-06