Software Engineer, AI/ML (Infrastructure & Platform)

Build core AI infrastructure

Build AI capability layers (tools / skills)

• Design and implement tools (“skills”) that agents and applications rely on, including APIs, workflows, and integrations
• Define clear interfaces for capabilities such as data retrieval, calculations, document processing, and external system actions
• Build reusable, composable abstractions that enable safe and scalable tool usage across systems
• Ensure tools are reliable, observable, and secure, especially when interacting with sensitive data

Enable agentic systems at scale

Develop evaluation and observability systems

Own reliability and performance

Build data and retrieval infrastructure

Create internal platforms and developer tooling

Work closely with product and AI teams

Qualifications

• A degree in Computer Science, Engineering, or a related quantitative field (or equivalent practical experience)
• Strong software engineering fundamentals, including system design, distributed systems, and writing maintainable code
• Proven track record of building and operating production systems at scale
• Proficiency in Python, TypeScript, C#, and comfort working across a polyglot stack, picking up new languages and frameworks as needed
• Experience building backend systems, APIs, or infrastructure platforms
• Experience working with AI/ML systems in production, including LLM integrations or data pipelines
• Experience designing or integrating systems with tool/skill abstractions (e.g., function calling, APIs, or capability layers used by AI systems)
• Ability to operate in ambiguous, fast-moving environments with high ownership

Preferred Qualifications (Bonus Points)

• Experience building AI platforms or infrastructure layers (not just applications)
• Experience with: RAG systems, vector databases (e.g., Pinecone, Weaviate, pgvector) Agent orchestration frameworks (e.g., LangGraph, LangChain, or custom systems) Evaluation and observability tooling for AI systems
• RAG systems, vector databases (e.g., Pinecone, Weaviate, pgvector)
• Agent orchestration frameworks (e.g., LangGraph, LangChain, or custom systems)
• Evaluation and observability tooling for AI systems
• Experience designing or building tooling layers (skills/capabilities) for AI systems
• Experience designing scalable distributed systems or platform abstractions
• Experience with cloud infrastructure such as: GCP (Cloud Run) or AWS (ECS, Lambda) Containerized or serverless deployments
• GCP (Cloud Run) or AWS (ECS, Lambda)
• Containerized or serverless deployments
• Experience with event-driven systems, queues, and async processing
• Experience with MLOps, CI/CD, and production monitoring
• Experience working in regulated domains (LegalTech, FinTech, HealthTech)
• Familiarity with data privacy and security techniques (e.g., PII handling, redaction)

You Might Be a Fit If

You Might Not Be a Fit If

Benefits & Perks

• Competitive salary.
• Hybrid work in the New York area
• Excellent medical, dental, and vision insurance options, with low-cost premium structures that demonstrate our commitment to offering great value to our employees.
• 100% company-paid basic life insurance, short-term and long-term disability insurance.
• 100% paid parental leave upon eligibility.
• Company equity managed through Carta.
• 401k with match and 100% vesting upon hire.
• Flexible PTO in an environment where taking time off to relax or recharge is supported and encouraged.
• Take time off for holidays—and yes, your birthday counts too. Celebrate, relax, and recharge without thinking twice.
• Wealth is an equal opportunity employer and encourages people from all backgrounds to apply. Should you have a disability or special need that requires accommodation, please let us know.