Machine Learning Engineer (LLM/ML Infrastructure)

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.

Responsibilities:

• Design and build ML infrastructure, training pipelines, and model deployment systems

Searching for Machine Learning & AI roles that provide visa sponsorship? Connect with international employers through Machine Learning & AI Jobs with Visa Sponsorship opportunities actively seeking talented professionals.

• Develop SDKs and tools for AI agent development
• Work closely with customers to deploy and optimize AI solutions
• Improve model performance, reliability, and scalability

Requirements:

• 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

Interested in opportunities specifically in United State? Discover our dedicated Visa Sponsorship Jobs in United State page featuring roles from top employers in this location.

• Interest in ML infrastructure and startup environment

Nice to have:

• 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