Machine Learning Engineering Team Lead

The role

You will lead the machine learning team to improve model quality and consistency across various operating conditions. You will also support edge model development and collaborate with software, robotics, hardware, and product teams.

What you will do

• Improve the quality and consistency of models across a growing breadth of operating conditions
• Introduce feedback loops and ML Ops to accelerate model improvement
• Support exploration of model deployment on devices for real-time use cases
• Lead and grow the machine learning team
• Collaborate with Software, Robotics, Hardware and Product teams
• Drive architecture and approach for new and existing models
• Lead development of training and inference pipelines

Required qualifications

• Experience in early-stage or fast-moving environments where scope and ownership evolve quickly
• Strong familiarity with models built on images, video, lidar, pointclouds, speech, or geospatial data sources
• Building and using cloud-based training and inference pipelines
• 3+ years experience in leadership: setting clear direction, driving architecture, reviewing work, and mentoring
• Experience creating, curating, and cleaning training datasets
• Building evaluation and feedback loops that drive model improvement
• Strong programming skills
• Expertise in deep learning frameworks such as PyTorch, scikit-learn, and TensorFlow
• Excellent communication and collaboration skills

Nice to have

• Experience developing models for and deploying models to run on edge-devices
• Experience with deep learning training frameworks such as MLflow, Lightning, Weights and Biases, or similar
• Experience with semantic mapping, image processing techniques and/or computer vision fundamentals
• Experience with Apache Airflow (especially 3.x) and/or Astronomer
• Experience with one or more of: Sensor Fusion, SLAM, Visual Odometry
• 8+ years of professional experience

Benefits

• Competitive salary and equity package
• Comprehensive medical, dental, and vision coverage for you and your eligible dependents
• Life insurance and short- and long-term disability coverage
• 16 weeks of fully paid parental leave to support all new parents
• Flexible, unlimited paid time off
• 401(k) retirement savings plan
• Free OneMedical membership
• Commuter benefits
• Snacks, goodies, and team lunches provided twice a week

About the company

Treeswift empowers energy companies to modernize their field work using physical AI, hardware, sensors, and software. The company is headquartered in midtown Manhattan with additional offices in San Francisco and Philadelphia.