AI Scientist
Xaira Therapeutics
PyTorch
Senior Staff Machine Learning Engineer – Foundational Models
Stack AV
full-time
Posted on:
Location Type: Remote
Location: Remote • Pennsylvania • 🇺🇸 United States
Visit company websiteJob Level
Senior
Tech Stack
GraphQLOpen SourcePythonPyTorchSQLTensorflow
About the role
- Own the technical direction of the team in leveraging foundational models and related techniques for the dual mission of the team.
- 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 senior leadership team to prioritization of resolutions.
- Align stakeholders with strong presentation and communication skills.
- Lead challenging technical topics and build consensus.
Requirements
- BS (with 9 years of experience), MS (with 8 years of experience) or PhD (with 6 years of experience) in developing machine learning models or systems.
- Track record of driving multi-department 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 environment.
- 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: 1-2 year of experience with people management.
- Nice to have: Experience with containerization, FastAPI, REST /GraphQL, SQL.
- Nice to have: Experience with autonomous driving and/or robotics.
Benefits
- We are proud to be an equal opportunity workplace.
- We believe that diverse teams produce the best ideas and outcomes.
- We are committed to building a culture of inclusion, entrepreneurship, and innovation across gender, race, age, sexual orientation, religion, disability, and identity.
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 skillsleadershipconsensus buildingproblem definitionexperimentationiterationproject management
