ML Infrastructure Engineer
xAIGenerative AI company
Palo Alto, United States$180K - $440KSenior
Valor Equity Partners
Andreessen Horowitz
Sequoia Capital
Vy Capital
Fidelity Management & Research Company
Prince Alwaleed Bin Talal
Data & AI
About the role
TL;DR
Optimize ML platform for high-performance and scalability.
- •As an ML Infrastructure Engineer, you will play a pivotal role in building and optimizing the reliable, high-performance ML platform that powers recommendations on X.
- •Key Responsibilities Designing, building, and scaling GPU compute infrastructure, training frameworks, and experimentation tools to enable rapid iteration on ML hypotheses Developing data pipelines and integrating large-scale data, training, and inference systems Collaborating with ML teams to productionize models and ensure seamless integration across the stack Requirements Bachelor, Master, Post-graduate or PhD in computer science, machine learning, or other quantitative discipline; or equivalent work experience 2+ years of industry experience working with high traffic or large-scale production environments, distributed systems, GPU infrastructure, and/or deep learning applications 2+ years experience with ML platforms, training infrastructure, or close collaboration with modeling engineers and data scientists Strong proficiency with Python and experience with compiled languages such as C++ or Rust
Required skills
PythonC++Rust
Nice-to-have skills
JAXPyTorchDockerKubernetesCI/CDLinuxAnsible
Domain expertise
ai
Benefits & perks
equity, comprehensive medical, vision, and dental coverage, 401(k) retirement plan, short & long-term disability insurance, life insurance, various other discounts and perks
Tech stack
PythonC++RustJAXPyTorchDockerKubernetesCI/CDLinuxAnsible