Senior Software Engineer - ML Services
TLDR
Build AI-powered, low-latency ML services for complex document understanding, advancing agentic systems and production-grade infrastructure.
Be a driving force for the team's agentic coding practice - establishing patterns for spec-driven development, context engineering, and multi-agent orchestration that compound team output.
Design, develop, and maintain AI-powered applications and services, focusing on high availability, performance, and reliability.
Work closely with Machine Learning Engineers, building AI capabilities from early R&D prototyping to scalable operation in multiple production stacks.
Build ML services designed for low-latency or high-throughput, and optimized for high cost efficiency.
Lead cross-functional initiatives, collaborating with Machine Learning Engineers, Product Managers, and other teams to align technical solutions with business needs.
Drive innovation by making strategic technical decisions, setting best practices, and mentoring engineers to elevate the team’s technical bar.
Improve ML development process by enhancing the internal ML platform, facilitating data sourcing, and enabling ML observability for deployed services.
Fluency with agentic coding tools (Claude Code, Codex or equivalent). Experience writing effective specs, managing repo context, and reviewing/verifying agent output.
5+ years of experience in software development, with a strong backend and ML engineering background.
Experience working with cloud native technologies (Docker, Kubernetes).
Experience working with cloud platforms (AWS, GCP, or Azure).
Experience with building ML systems, from early prototype, to operating at scale in production.
Strong software architecture skills, with experience designing and scaling distributed systems and cloud-based applications.
Experience mentoring engineers and contributing to team-wide technical direction.
Strong communication skills, with the ability to translate technical challenges into business impact.
Experience with ML frameworks like pytorch.
Experience with serving ML models and LLMs using frameworks like vllm.
Experience with developer-facing products and building intuitive APIs.
Benefits
Education Stipend
Lifelong Learning – An education stipend to help you grow and keep learning.
Equity Compensation
Stock Awards – We want you to share in our success with equity grants.
Health Insurance
Comprehensive Healthcare – Medical, dental, and vision coverage for you and your dependents.
Wellness Stipend
Wellness Stipend – Support for your health and well-being, whether it’s fitness programs or self-care.
AI Fund is a venture studio that collaborates with entrepreneurs to develop and launch innovative companies from the ground up. By leveraging resources from top-tier venture capital firms and Fortune 500 companies, we provide the support necessary to take ideas from inception to market with confidence.