AI/ML Engineer

As an AI/ML Engineer at Reflective, you will be the technical lead for ML architecture and MLOps for our multi-scale modeling project, turning petabytes of climate and microphysics data into robust, lightning-fast surrogate models.

• Design, implement, and benchmark surrogate architectures for atmospheric and aerosol processes using best-in-class AI/ML methodologies.
• Build reproducible data pre-processing and training pipelines (PyTorch, JAX, Lightning, Weights & Biases).
• Design and implement state-of-the-art computer vision models, including Vision Transformers (ViTs) and generative approaches like diffusion, to extract insights and build surrogates from complex climate and microphysics datasets.
• Automate hyper-parameter sweeps and uncertainty calibration on multi-GPU clusters; monitor cost vs. accuracy trade-offs.
• Containerize and document workflows so scientists can retrain or fine-tune models with minimal friction.
• Collaborate on releases, tutorials, and user-support channels; respond rapidly to bug reports

The purpose of this project is to prototype surrogate models, machine-learned emulators trained on targeted high-resolution simulations, designed to increase the accuracy and speed of modeling SAI deployment scenarios to help make scientifically-informed decisions about sunlight reflection for climate impact mitigation.

• You have 3+ years in applied ML engineering
• You have developed or contributed to at least one end-to-end deep-learning system that is in production or used for research purposes.
• Strong software-engineering skills (Python, Linux, GitHub Actions, Docker) and comfort with distributed training on GPUs.
• Familiarity with scientific data formats (NetCDF, Zarr) and geospatial arrays.
• Ability to explain complex ML choices to domain scientists and incorporate their feedback.
• You’re creative and attached to outcomes, not process – you’re constantly looking for new paths to the destination and excited to switch gears if there’s a faster, better way to get something done.
• You are passionate about Reflective’s mission.

We expect this position to be a regular, full-time position, with an annual salary between $145,000 and $210,000 USD, depending on level of experience. In addition to salary, we offer a comprehensive set of benefits to all full-time employees:

• Medical, dental, vision insurance
• 401(k)
• Professional and personal development
• Generous paid time off and sick leave, including 12 weeks paid parental leave
• Flexible working hours

Reflective is a philanthropically-funded initiative to develop the necessary knowledge base and do the requisite technology research and development, urgently and responsibly, for sunlight reflection as a climate impact mitigation option alongside emissions reductions, adaptation, and carbon removal.