Machine Learning Research Engineer

LLNL•Livermore, CA

About The Position

We have an opening for Machine Learning Research experts to join our team and advance the discipline as well as apply cutting edge tools and techniques to some of society’s most important problems. You will work with or lead a multi-disciplinary team consisting of machine learning experts, data science practitioners, and domain scientists in areas ranging from fundamental research in machine learning, i.e., AI safety, robustness, uncertainty quantification, or interpretability to applied problems in fields such as high energy density physics, material science, predictive medicine, and treatment discovery. You will also have the opportunity develop and lead independent research thrust and engage with a variety of related research projects in parallel computing, data analysis and visualization, or applied mathematics. This position is in the Center for Applied Scientific Computing (CASC) Division within the Computing Directorate.

Requirements

M.S. in Computer Science, Applied Mathematics, Statistics or related field or the equivalent combination of education and related experience.
Experience in at least one machine learning research area, such as, foundation models, representation learning, safety & robustness, uncertainty quantification, interpretability, physics-constrained ML, or graph-based learning as demonstrated in software artifacts or publications at high impact AI/AL focused venues.
Experience developing, implementing, and applying advanced statistical or machine learning models and algorithms using modern software libraries such as PyTorch, TensorFlow, or similar as evidence through medium to large scale deep learning models and experiments.
Experience in working with diverse teams to solve complex problems and deliver practical solutions.
Comprehensive analytical and problem-solving skills necessary to craft creative solutions and solve complex problems.
Ph.D. in Computer Science, Applied Mathematics, Statistics or related field or the equivalent combination of education and related experience.
Demonstrated research productivity, as documented by publications, reports, presentations, and/or open-source software in high impact AI/ML focused venues, such as, NeurIPS, ICML, ICLR, CVPR, AAAI, AISTATS, UAI, KDD, or JMLR
Advanced verbal and written communication skills necessary to interact with a multi-disciplinary research team, author technical and scientific reports and papers, and deliver scientific presentations.

Nice To Haves

Experience with high-performance computing, GPU programming, parallel programming, cloud computing, and/or related methods including running numerical simulations or complex workflows.
Experience in working with subject matter experts in one or more areas, such as physics, biology, or engineering.
Background in statistics, applied mathematics, or related area.

Responsibilities

Research, develop, implement, and evaluate new machine learning techniques for multiple applications in a collaborative scientific environment.
Adapt and deploy common machine learning software stack on large-scale high performance computing clusters.
Actively participate with project scientists and engineers in defining, planning, and formulating experimental, modeling, and simulation efforts for complex problems stemming from national security applications.
Adapt current machine learning research to real world applications at scale, with potentially limited and noisy data, with a high consequence of error, and guide the development of practical solutions.
Collaborate with a broad spectrum of scientists and engineers, internally and externally, to accomplish research goals.
Perform other duties as assigned.
Provide guidance to subject matter experts in various fields to jointly explore the potential for machine learning research to solve domain specific challenges.
Establish future research directions and author grant proposals including presentations to programmatic sponsors and external funding agencies.
Lead small to mid-sized research teams in theoretical or applied machine learning in support of one or more mission related scientific applications.
Present and disseminate research results at scientific conferences and in peer-reviewed publications.