Machine Learning Infrastructure Engineer
WindBorne SystemsWeather Intelligence company
Redwood City, United States$140K–$240KMid
Khosla Ventures
Pear VC
Susa Ventures
Footwork VC
Convective Capital
Ubiquity Ventures
Data & AI
About the role
TL;DR
Build and maintain ML infrastructure for real-time weather forecasting models.
- •WindBorne Systems is seeking a Machine Learning Infrastructure Engineer to optimize their AI weather models.
- •This role will focus on improving the operationalization and infrastructure supporting these models, allowing researchers to concentrate on development rather than firefighting.
- •Key Responsibilities Own Research to Operations pipelines, ensuring real-time forecast delivery with low latency.
- •Manage inference scaling and compute strategy across on-prem and cloud environments.
- •Build and maintain reliable data pipelines for training and real-time data, handling upstream complexities and data quality checks.
- •Enhance training infrastructure for reliability and auto-recovery of distributed training jobs.
- •Requirements Experience running production ML systems and building robust infrastructure.
- •Experience with large datasets and managing custom deployments for fast-paced model releases.
- •Proficiency with PyTorch, Docker, memory management, compression, and network debugging.
- •A strong affinity for systems and structure to balance research team chaos.
Required skills
PythonPyTorchDockerKubernetesAWSGoogle CloudBashGit
Domain expertise
climatetechaerospace
Benefits & perks
401(k), Dental, health, and vision insurance, Unlimited PTO, Stock Option Plan, Office food and beverages
Tech stack
PythonPyTorchDockerKubernetesAWSGoogle CloudBashGit