Metropolis

Senior Machine Learning Engineer, Computer Vision

Metropolis
Apply
3 months ago

Base Salary

$150k - $200k/yr

Responsibilities

  • Design, develop, and deploy computer vision models for object detection, tracking, OCR, image search, and scene understanding.
  • Build and optimize deep learning models for accuracy, performance, scalability, and production deployment.
  • Integrate multimodal approaches using visual, textual, and other data modalities.
  • Collaborate with data engineers and software engineers to deliver end-to-end solutions.
  • Lead scalable pipelines for data processing, model training, and model deployment.
  • Optimize models for CPUs, GPUs, edge devices, cloud platforms, and mobile platforms.
  • Conduct experimentation and A/B testing to validate model effectiveness and business alignment.
  • Mentor junior team members and provide technical guidance.
  • Write clean, efficient, maintainable code following software engineering and machine learning best practices.

Requirements

  • PhD in Computer Science, Engineering, or a related field, or equivalent work experience.
  • At least five years of hands-on machine learning and computer vision experience, including deploying models into production.
  • Proficiency in Python and ML frameworks including PyTorch, TensorFlow, ONNX, and TensorRT.
  • Experience with model optimization such as quantization and pruning and deployment across cloud, edge, or mobile platforms.
  • Familiarity with AWS, GCP, or Azure; Docker; and ECS or Kubernetes.
  • Experience building and maintaining data pipelines such as Airflow.
  • Understanding of agile development processes and CI/CD pipelines and tools such as GitHub Actions and Jenkins.
  • Strong communication skills and the ability to present complex technical information clearly.
  • C++ experience, experience in high-growth environments, and publications in CVPR, ICCV, or NeurIPS are preferred.

Benefits

  • Office-first role requiring at least four days per week on-site.
  • Healthcare benefits, 401(k), short- and long-term disability coverage, basic life insurance, stock option plan, and bonus plans may be available.
  • Total compensation may include benefits beyond base salary.

Tech Stack

Apache AirflowAWSAzureC++DockerGitHub ActionsGoogle Cloud PlatformJenkinsKubernetesPythonPyTorchTensorFlow

Categories