Lead Machine Learning Engineer (Agentic Infrastructure)

Job Description

Over the past 8 years, Codesearch AI have had the pleasure to work with some of the most ground-breaking and successful starts ups around. We can safely say this company is as exciting as it gets.

We are an exclusive partner to a YC-backed start-up that's building truly transformative AI technology. Their agentic AI platform goes well beyond chat interfaces, offering ground-breaking memory capabilities that solve real enterprise problems with unprecedented accuracy. As validation of their innovative approach, one of the world's most widely used AI tools is already exploring adoption of their technology.

With a founding team of accomplished researchers and engineers from organizations like LinkedIn and FAIR, they're now expanding their core team to bring this revolutionary product to market.

The Role

They're seeking their first dedicated ML Engineer to help productise their Agentic AI platform. This role is perfect for someone who loves to move fast, ship usable systems, and operate at the intersection of LLMs, infrastructure, and software engineering.

What You'll Be Doing

• Take working prototypes of LLM-based agents and productize them into scalable, robust systems

• Build infrastructure and pipelines to support and integrate AI Agents in real-world enterprise environments

• Collaborate with the founding team to integrate models into internal and external user flows

• Write clean, production-ready code - often improving or refactoring existing prototypes

• Think holistically aboutagent lifecycle, observability, failure handling, and scalability

• Help define thetech stack and architecturefor core components of the platform

Contribute to novel research and publish at top conferences when opportunities arise

What You'll Have

• MSc or PhD in Machine Learning, Computer Science or a related field

• 5+ years of experience in ML engineering, MLOps and/or backend/infra-focused roles

• Experience integrating LLMs into enterprise SaaS or internal tooling

• Strong Python experience with ML/LLM libraries (e.g., Transformers, LangChain, LangGraph, OpenAI APIs)

• Experience with cloud platforms (AWS, GCP, or Azure), deployment, and CI/CD pipelines

• Familiarity with containerization (Docker, Kubernetes) and observability (e.g., Prometheus, Grafana)

• A builder mindset: you're comfortable with ambiguous specs, early-stage infrastructure, and iterating fast

• Excellent communication and self-management skills

Nice To Have

• Familiarity with agentic frameworks, orchestration tools, or vector databases

• Background in DevOps/MLOps or platform engineering

• Passion for building something from scratch and seeing the impact of your work in production

What We Offer

• Competitive salary with equity options based on experience and profile
• Flexible work arrangements with remote/hybrid options
• Comprehensive health benefits and wellness programs
• Professional development budget for conferences and continued learning
• A front-row seat to the agentic AI evolution
• Full ownership and trust over your code and system decisions
• A lean, expert team with direct access to product, users, and strategic investors

Opportunity to shape the future of AI in a fast-growing market segment

#J-18808-Ljbffr