Machine Learning Engineer

Role: Machine Learning Engineer

Location:

Los Altos, CA (2 days onsite Tue & Wed mandatory)
Duration:

6+ months
Requirements:

4+ years of professional Client engineering experience at an AI/Client-focused organization.
Familiarity with the state-of-the-art in behavior learning, language, and/or computer vision.
Experience training large-scale foundation models (VLMs, text-to-video models, etc) utilizing distributed training and high-performance optimization techniques such as quantization, mixed precision, model parallelism, data parallelism or FSDP.
Extensive practical experience with PyTorch.
Strong proficiency in Python and software development best practices such as unit testing, documentation, code review, continuous integration, and dependency management.
Familiarity with data pipelines, model serving and optimization, cloud training, and dataset management.
An ability to move fast and switch between modes of rapid prototyping and robust implementation as required.
Nice to Haves:

Experience deploying models on embodied systems/robots.
Experience working in mixed teams of research scientists and engineers.
Experience with Amazon EC2, S3, and/or Sagemaker.
Experience with Bazel.

#J-18808-Ljbffr