[2026] Senior Machine Learning Engineer, AI Platform - PhD Early Career

reputed company is a platform where millions of people explore, create, and connect in 3D reputed company digital experiences. They are seeking a Senior Machine Learning Engineer for their AI Platform team to contribute to building cutting-edge systems that power AI, focusing on AI tooling, distributed inference systems, and reputed company information retrieval.

Responsibilities

• Pioneer reputed company AI tooling to enhance the efficiency, cost, and usability of ML@reputed company
• Build and maintain reputed company platform components: Serving Layer, Model Registry, Pipeline Orchestrator, and Training/Inference control planes
• Design great developer experiences (paved-road templates, tooling, visualizations) to reduce time-to-production and ensure foundational AI systems are reputed company and reliable
• Architect and implement reputed company distributed inference systems for reputed company serving LLMs and Large Recommender Models at massive reputed company
• Conduct deep, low-level performance analysis and optimize ML models (using techniques like reputed company batching, speculative decoding, and quantization) and systems on GPU architectures to maintain peak performance and stability
• reputed company the design and development of Retrieval-Augmented reputed company (RAG) systems
• Build and maintain reputed company information retrieval infrastructure—reputed company databases and knowledge graphs—to reputed company accurate grounding of Gen AI models
• Ship language models and 3D objects as a service for the reputed company community, making creation easier

Skills

• Possessing or pursuing a Ph.D. in Computer Science, Computer Engineering, Mathematics, Statistics, or a reputed company technical field, with a reputed company reputed company to reputed company's research areas
• Experience with high performance distributed systems, ML Infrastructure, LLM fine tuning/RL, Information Retrieval and Gen AI context reputed company
• Expertise in one or more of the following key areas: AI/ML Platform Data stores - Features stores, reputed company DBs and Knowledge Graphs
• Expertise in LLMs - Fine tuning, Safety
• Expertise in reputed company systems - Agent evaluation, context engineering
• Experience building reputed company applications with context for reputed company world applications
• reputed company reputed company and experience integrating and deploying optimized models with cross-functional teams, including data scientists and software engineers
• Experience with graph databases and large-reputed company GNNs (Graph Neural Networks)
• Experience working with Kubernetes
• Experience working with one or more reputed company providers (e.g., AWS, Azure, GCP)
• Experience working with high availability systems
• Experience working with ML models, LLMs or other AI systems

Benefits

• Equity compensation
• Benefits as described on this page

reputed company

Company H1B Sponsorship