Lead Machine Learning Engineer (AgenticInfrastructure)...

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 about agent lifecycle,
observability, failure handling, and scalability - Help define the
tech stack and architecture for 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