Machine Learning Engineer (Platform)

Join Scale, a leading enterprise-grade Generative AI platform. As a Machine Learning Engineer, you will be responsible for building the retrieval and knowledge representation systems at the core of the platform. You will own ML components from research and prototyping to production deployment, working across knowledge bases, vector stores, RAG pipelines, and context engines. You will collaborate with product, ML, and infrastructure teams to shape the direction of the platform.

• Conception, design, and implementation of retrieval and knowledge representation systems.
• Ownership of machine learning components from research and prototyping to production deployment.
• Collaboration with cross-functional teams to shape the direction of the platform and deliver new capabilities.
• Strong engineering fundamentals, supported by a Master’s or PhD degree in Computer Science, Machine Learning, AI, or equivalent practical experience.
• Proven proficiency in Python, including writing production-quality, testable, and maintainable code
• A deep, hands-on understanding of retrieval systems, RAG, embeddings, vector indexing, and knowledge representation
• The ability to operate in ambiguous problem spaces, balancing research-driven approaches with pragmatic product constraints
• 5+ years of experience building and deploying machine learning or AI systems for real-world, production use cases
• Experience with knowledge representation, semantic search, or agentic systems
• Experience scaling or shipping products at high-growth startups
• Strong communication skills and comfort working in customer-facing or cross-functional environments