Stack AV

Staff Machine Learning Engineer – Foundational Models

Stack AV

full-time

Posted on:

Location Type: Remote

Location: Remote • Pennsylvania • 🇺🇸 United States

Visit company website
AI Apply
Apply

Job Level

Lead

Tech Stack

GraphQLOpen SourcePythonPyTorchSQLTensorflow

About the role

  • Contribute to the technical direction of the team and broader perception org on leveraging foundational models and related techniques.
  • Design and implement offboard pipelines to train, distill, evaluate and serve Machine Learning Models.
  • Demonstrate that these pipelines work with open source data and contribute to open-source libraries.
  • Proactively identify opportunities for improvements and work with leadership team on prioritization of resolutions.
  • Align stakeholders with strong presentation and communication skills.
  • Lead challenging technical topics and build consensus.

Requirements

  • BS (with 6 years of experience), MS (with 5 years of experience) or PhD (with 3 years of experience) in developing machine learning models or systems.
  • Track record of driving projects from start to completion, including conception, problem definition, experimentation, iteration, and finally productization.
  • Experience in contributing to or leveraging one or more of the recent advances related to foundational models such as pre-training models, RL based fine-tuning , neural rendering, diffusion based synthetic image/video generation, World Foundational Models.
  • Experience with processing large datasets, training ML models in distributed environments.
  • Track record of developing end to end pipelines for training Machine Learning Models
  • Strong experience in Python and at least one of popular Deep Learning frameworks like Pytorch, Tensorflow, JAX.
  • Nice to have: Experience with containerization, FastAPI, REST /GraphQL, SQL.
  • Nice to have: Experience with autonomous driving and/or robotics.
Benefits
  • Health insurance
  • Paid time off

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
machine learning modelsfoundational modelspre-training modelsreinforcement learningneural renderingdiffusion based synthetic image generationend to end pipelinesPythonPytorchTensorflow
Soft skills
presentation skillscommunication skillsleadershipproblem definitionconsensus building
Certifications
BSMSPhD
Gainwell Technologies

Principal AI Engineer

Gainwell Technologies
Leadfull-timeTexas · 🇺🇸 United States
Posted: 2 hours agoSource: jobs.gainwelltechnologies.com
AWSCloudDockerKubernetesPythonTerraform
3Pillar Global

AI Engineer

3Pillar Global
Mid · Seniorfull-time$80k–$110k / year🇺🇸 United States
Posted: 5 hours agoSource: jobs.lever.co
JavaScriptNeo4jNext.jsPythonReactTypeScript
Gopuff

AI Engineer III

Gopuff
Senior · Leadfull-time$215k–$275k / year🇺🇸 United States
Posted: 6 hours agoSource: jobs.lever.co
CloudPyTorch
Gopuff

AI Engineer II

Gopuff
Mid · Seniorfull-time$175k–$220k / year🇺🇸 United States
Posted: 6 hours agoSource: jobs.lever.co
CloudPyTorch