Machine Learning Engineer

Skills
Computer ScienceAcademic PublishingComputer VisionDatasetsDeep LearningPattern RecognitionPytorch
Role

What the job involves

The main requirements, responsibilities and hiring steps.

Requirements

  • 5+ years experience in related field
  • Experience with tensorflow/pytorch or Jax
  • Experience building distributed, scalable software
  • Proficiency in Python
  • Ability to thrive in fast-paced environment
  • Communicate complex technical concepts
  • Thoughtfulness and detail-orientation

Nice to have

  • Master or Ph.D. in related field
  • Experience with research papers
  • Computer vision experience

Day to day

  • Collaborate with research engineers to build infrastructure for research operations and automate ML model training and optimization in computer vision
  • Write frameworks with modular components to prototype trainings and implement cutting-edge research ideas
  • Build robust ML pipelines for inference on large datasets, optimizing GPU use, memory, speed, and accuracy, and design APIs to automate training and validation for specialized models