Demo

Machine Learning Engineer

Karkidi
Berkeley, CA Full Time
POSTED ON 1/26/2025
AVAILABLE BEFORE 4/21/2025

The Opportunity

We’re looking for a driven machine learning engineer comfortable working on large integrated machine learning systems. Experience with robots or other embodied systems (such as autonomous vehicles) is a bonus.

If our mission of revolutionizing robotics through machine learning resonates with you, get in touch and let’s talk about how we can create the next generation of AI-powered capable robots together.

Responsibilities

  • Collaborate with internal research scientists and our partner labs at top academic research universities, including MIT, Stanford, Berkeley, CMU, Columbia, and Princeton to drive pioneering research at scale.
  • Build, improve, and robustify end-to-end integrated ML pipelines for training multimodal (language, images, video, actions) models at scale.
  • Train, finetune, and serve robot foundation models with a strong MLOps mindset.
  • Build processes for integrating collaboration-produced and open-source advancements and code into our internal stack.
  • Build and improve large data pipelines for foundation model training.
  • Be a key member of the team and play a critical role in rapid progress measured by both the development of internal capabilities and high-impact external publications.

Qualifications

  • 2 years of professional ML engineering experience at an AI / ML-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 Amazon EC2, S3, and / or Sagemaker.
  • Experience with Bazel.
  • The pay range for this position at commencement of employment is expected to be between $165,760 and $238,280 / year for California-based roles; however, base pay offered may vary depending on multiple individualized factors, including market location, job-related knowledge, skills, and experience. Note that TRI offers a generous benefits package (including 401(k) eligibility and various paid time off benefits, such as vacation, sick time, and parental leave) and an annual cash bonus structure. Details of participation in these benefit plans will be provided if an employee receives an offer of employment.

    J-18808-Ljbffr

    Salary : $165,760 - $238,280

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Machine Learning Engineer?

    Sign up to receive alerts about other jobs on the Machine Learning Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $123,167 - $152,295
    Income Estimation: 
    $146,673 - $180,130
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $77,900 - $95,589
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $179,606 - $233,815
    Income Estimation: 
    $211,413 - $298,244
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Karkidi

    Karkidi
    Hired Organization Address Boston, MA Full Time
    We are looking for innovative and creative individuals who seize opportunities to uncover hidden drivers, impacts, and k...
    Karkidi
    Hired Organization Address Mountain View, CA Full Time
    Minimum qualifications : Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, a related ...
    Karkidi
    Hired Organization Address San Diego, CA Full Time
    The Digital Modernization Sector has a career opportunity for a Senior Solution Architect for Innovations supporting the...
    Karkidi
    Hired Organization Address Mississippi State, MS Full Time
    You will join a team working with next gen technologies on strategic innovation projects in order to identify areas for ...

    Not the job you're looking for? Here are some other Machine Learning Engineer jobs in the Berkeley, CA area that may be a better fit.

    Machine Learning Engineer

    Realm, Hayward, CA

    Machine Learning Engineer

    SwingVision, Berkeley, CA

    AI Assistant is available now!

    Feel free to start your new journey!