Role overview
Machine Learning Engineer (LLM / ML Infrastructure)
We are a fast-growing AI infrastructure startup backed by top-tier VCs, building agentic AI systems that enable developers to build and optimize AI agents using natural language.
We are looking for a Machine Learning Engineer with strong interest in ML infrastructure and large language models.
What we're looking for
- Design and build ML infrastructure, training pipelines, and model deployment systems
- Develop SDKs and tools for AI agent development
- Work closely with customers to deploy and optimize AI solutions
- Improve model performance, reliability, and scalability
- PhD in Machine Learning, AI, NLP, or related field
- Strong experience with Python and ML frameworks (PyTorch, TensorFlow, or JAX)
- Background in LLM, NLP, or deep learning
- Interest in ML infrastructure and startup environment
- Publications in top AI conferences (NeurIPS, ICML, ICLR, ACL, EMNLP)
- Experience building ML platforms, APIs, or SDKs
Location: Mountain View, CA (4+ days onsite)
Visa sponsorship available
Tags & focus areas
Used for matching and alerts on DevFound Fulltime Ai Machine Learning Deep Learning Nlp Mlops Pytorch Tensorflow Generative Ai