Sr Software Engineer - ML Infrastructure

Job Description

Build reusable ML frameworks, distributed execution capabilities, and developer tools for model training, simulation, evaluation, and productionization. Lead systems from design through rollout and operations, improve performance and reliability, and collaborate with modeling and platform teams.

Requirements: Requires at least 6 years of software engineering experience, strong Python skills, and experience with ML infrastructure, distributed systems, and production ML workflows. Candidates should understand performance fundamentals, have delivered systems into production, communicate technical tradeoffs effectively, and hold a bachelor's or master's degree in a relevant field or equivalent practical experience.

Key Skills: Python, Machine Learning Infrastructure, Distributed Systems, Workflow Execution, Model Training, Batch Inference, Model Evaluation, Model Deployment, Ray, PyTorch, PyTorch Lightning, LightGBM, GPU Performance Profiling, Hyperparameter Tuning, Experiment Tracking, Workflow Orchestration

Benefits: Bonus Program, Equity Award, 401(k) Plan, Benefits