Senior Machine Learning Engineer

MyAIO; Los Angeles (CA); On-site; USD 150,000 - 220,000; Full time

Job summary

MyAIO in Los Angeles, CA is seeking a Senior Machine Learning Engineer to design and scale ML systems that power intelligent agents. You will own end-to-end components including LLM-based reasoning, retrieval, memory, and data pipelines for training and evaluation.

We expect 5+ years in ML or applied AI, deep knowledge of LLMs, embeddings, and vector search, and hands-on experience with PyTorch, LangChain, Ray, and Weaviate.

Qualifications

• 5+ years of experience in machine learning or applied AI, ideally including production-level deployments.
• Deep understanding of LLMs, embeddings, and vector search.
• Experience with PyTorch, LangChain, Ray, and Weaviate.
• Strong engineering mindset focused on shipping, testing, and iterating.

Responsibilities

• Design and optimize LLM-based systems powering the agent core
• Prototype advanced retrieval, planning, and memory systems for AI agents
• Build and own data pipelines for training, fine-tuning, and evaluation
• Collaborate with product and infrastructure teams to deploy scalable ML features

Skills

5+ years experience

LLMs

data pipelines

Education

Bachelor-level education

Tools

PyTorch

LangChain

Ray

Weaviate

Job description

Build and scale machine learning systems for intelligent agents in Los Angeles, CA onsite.

Responsibilities

• Design and optimize LLM-based systems powering the agent core
• Prototype advanced retrieval, planning, and memory systems for AI agents
• Build and own data pipelines for training, fine-tuning, and evaluation
• Collaborate with product and infrastructure teams to deploy scalable ML features

Requirements

• 5+ years of experience in machine learning or applied AI, ideally including production-level deployments
• Deep understanding of LLMs, embeddings, and vector search
• Experience with frameworks and tools such as PyTorch, LangChain, Ray, and Weaviate
• Strong engineering mindset focused on shipping, testing, and iterating

Technologies

• LLMs
• Embeddings
• Vector search
• PyTorch
• LangChain
• Ray
• Weaviate

About the role: MyAIO is building the next generation of intelligent agents, with a Senior Machine Learning Engineer focused on the intersection of research and application. The role centers on architecting and scaling ML systems that support agent reasoning, planning, and communication, and on pushing model capabilities into real-world usage.