Finding the best job has never been easier
Apple Lead AI Engineer United States, West Virginia 98380954
Yesterday
Share
In this role, you will work on some of the most challenging problems at the intersection of deep learning, computer vision, LLM, Foundation Models, Agentic AI, and geospatial data. You will collaborate with a cross-functional team of engineers, ML scientists, and map specialists to build AI systems that understand the world at scale.
- Architect large complex Agentic AI systems and implement end-to-end integrating ML components and training pipelines, inference services, tools, evaluations and safety.
- Fine-tune and deploy large foundational models including LLMs, LVMs and LMMs for various real-world mapping tasks.
- Contribute to and scale our AI infrastructure, making it easier to experiment and iterate quickly.
- Evolve early-stage ideas into robust, production-ready systems collaborating with a large cross function team.
- BS with at least 5 years of machine learning or software engineering, with recent work on LLMs or generative AI.
- Strong software engineering fundamentals — you have shipped production systems and know how to build clean, robust, reliable and maintainable code.
- Shipped LLMs, VLMs, or other Generative AI systems at scale. Strong experience with ML and backend stack.
- Hands-on experience with processing large datasets, training ML models in distributed environments.
- Strong experience with Python, PyTorch, TensorFlow, JAX, containerization, FastAPI, REST/ GraphQL, Java and Scala.
- Strong interpersonal collaboration and communication skills, ability to work with high ambiguity and minimum supervision.
- MS or PhD in Computer Science, Artificial Intelligence, Machine Learning or related field with 10+ years of experience shipping ML models at scale.
- Experience working with spatial/geospatial data.
- Contributions to open-source ML/LLM tools or libraries.
- Experience with frameworks like LangGraph, MCP, or LlamaIndex.
- Understanding of retrieval-augmented generation (RAG), RLHF, multi-agent workflows, or LLMOps.
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
These jobs might be a good fit
