Staff Machine Learning Engineer – Foundational Models
Stack AV
full-time
Posted on:
Location Type: Remote
Location: Remote • Pennsylvania • 🇺🇸 United States
Visit company websiteJob 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