Altana - Senior Staff Machine Learning Engineer

Boston, MA; Brooklyn, NY; San Francisco, CA; Washington, D.C. - Hybrid$250k - $295k+ Equity2w ago

In Office Staff NA Cloud Computing Artificial Intelligence Machine Learning Engineer MLflow Databricks Git Swagger OpenAPI

Requirements

• Building multi-agent systems and agent tooling (e.g., LangGraph) • Hybrid retrieval: vector search, re-rankers, and embedding fine-tuning • Vision-language models for document understanding (invoices, bills of lading, customs forms) • Multilingual NLP • LLMOps: MLflow, Databricks Mosaic AI, and guardrails • Experience with safety, guardrails or auditability for AI in regulated or high-stakes domains (e.g., customs, legal, healthcare, finance) • Technologies we love • Languages: Python, Spark • Tools: Docker, Git, Kubernetes, Swagger/OpenAPI, AWS, Azure • Datastores: Databricks, OpenSearch, Postgres • ML Tools: PyTorch, Hugging Face, FAISS, MLflow • LLM Tools: vLLM, LangGraph, LangSmith/Braintrust, frontier model APIs • This role is remote or hybrid, based in the United States, UK, with the option to work from one of our hubs in New York City, Boston, Washington D.C., or LondonUS Salary Range and Benefits • This role is remote or hybrid, based in the United States, UK, with the option to work from one of our hubs in New York City, Boston, Washington D.C., or London

Responsibilities

• Drive the technical direction for agentic systems. Define the architecture and roadmap for the models and agentic systems that process trade data as it arrives (shipments, customs declarations, invoices, bills of lading) and turn it into risk, classification and compliance decisions in real time. • Build the pipeline from ingestion to decision. Lead the design of reliable, cost-efficient systems for ingestion, document understanding, entity resolution, inference and routing across models, at the scale of global trade. • Adapt models to trade. Use analyst feedback, production outcomes and agent traces to fine-tune and post-train models (SFT, DPO, RL with domain graders) where it measurably beats prompting, and keep every model improving over time. • Design the observability and feedback stack. Build the evals, tracing and feedback loops that decide what ships and catch regressions in production. • Lead beyond your team. Mentor senior engineers, raise the technical bar across Engineering, and help shape how we hire and level ML talent. • Shape the product and the business. Work with software engineers, data scientists, product managers and customers to generate new business ideas and turn ambiguous problems into shipped capabilities. • BS/MS in Computer Science, Machine Learning, or a related field, or equivalent experience • 10+ years building and shipping ML systems in production, with a proven track record of delivering product-oriented ML solutions, including recent work on LLM-powered products • Deep expertise in PyTorch and the modern Python ML stack (Hugging Face, scikit-learn, NumPy, Pandas/Polars) • Hands-on experience building production LLM applications: retrieval-augmented generation (RAG), tool/function calling, structured outputs, and agentic workflows • You've shipped agentic systems in production and know their failure modes: tool use, planning, state, recovery and human-in-the-loop • Experience post-training or adapting LLMs (SFT, DPO, RLHF/RLAIF, reward modeling) for a production use case • Rigorous about evaluation: you've built eval suites, golden datasets and LLM-as-judge pipelines • Sound judgment on when to use a frontier model API, a fine-tuned open-weight model, or a classical ML model, balancing cost, latency, and accuracy • Senior Staff-level technical leadership: you set architecture direction across teams, mentor engineers, and turn ambiguous problems into roadmaps • Passionate about applying SOTA techniques to real-world problems, with a willingness to get your hands dirty in the data and domain to achieve measurable outcomes • You care deeply about engineering excellence, supportability and knowledge-sharing • You have strong written and verbal communication skills • Heavy user of AI coding tools (Cursor, Claude Code) in your daily workflow

Benefits

• $250,000 - $295,000 USD • The salary range, to the extent specified for this role, is a good faith statement of the minimum and maximum levels of the annual based salary for the position. The base salary offered to a successful candidate will depend on a wide range of compensation factors, including, but not limited to, work experience, education and/or training, critical skills, and/or business considerations. Competitive equity grants are included in the majority of full time offers; and are considered part of Altana's total compensation package. Altana also offers either a discretionary bonus or a variable compensation plan depending on the role. Additionally, Altana offers top-tier benefits for full-time employees, including: • Flexible Time Off: Altana operates with a Flexible Time Oﬀ (FTO) policy that gives you agency over your own time oﬀ so you can maximize your work-life balance. • Parental Leave: We oﬀer industry leading Paid Parental Leave (PPL), providing 14 weeks of leave for non-birthing, adoptive, and foster parents and up to 26 weeks of leave for birthing parents, all paid at 100% of your base salary. • Health Benefits: We have a full suite of medical, vision, and dental benefits with generous employer contributions, designed to give you flexibility and choice for your individual health situation. Our high deductible health plan is 100% employer paid for employees and supplemented with an employer contribution to your Health Savings Account (HSA). There is also a Flexible Spending Account (FSA) option. • Supplemental Benefits: Altana provides life, short- and long-term disability, and AD&D insurance coverage, all at no cost to you, so you know that you and your loved ones are covered in case of an emergency. • 401(k) Savings: Save for and invest in your future using our Fidelity 401(k) retirement savings program. • Commuter Benefits: Save money on your commute by setting aside pre-tax funds for public transit or parking! • Wellness: Because we value mental and emotional health, every Altana employee has access to a free premium subscription to Calm, the #1 app for meditation, sleep, and mindfulness. • Pet Insurance: Pets are family too! Keep them healthy with Wishbone insurance and / or our Total Pet vet service and telehealth discount plan. • Employee Assistance Program: Free access to confidential personal support. • Dependent Care FSA: You will have access to a Dependent Care FSA, which allows you to set aside pre-tax funds for childcare expenses • The recruiter assigned to this role can share more information about the specific compensation and benefit details associated with this role during the hiring process. • Our Values • Our Values • Our values are the core beliefs that shape who we are, what we stand for, and how we behave.They form the foundation of Altana’s culture and integrity and guide how we hire, design, build, and connect with each other and our customers. • Trust: Our customers and partners entrust us with missions of the highest importance. We honor that by keeping our word, meeting commitments, and ensuring every action we take reinforces confidence in us. We rely on each other to deliver, to speak openly, and to hold ourselves accountable. • Trust: • Resilience: In a world of uncertainty and complexity, our work must withstand challenges, evolve with conditions, and ensure reliability over time. Resilience is both how we operate and what we deliver. It’s how we respond when things don’t go to plan –– we adapt, we support each other, and we keep moving forward. • Resilience: • Stewardship: We are stewards of every mission we touch. Because our work impacts lives and futures, we hold ourselves accountable to delivering mission impact and never compromising. Our responsibility extends beyond individual projects to the broader system of global trade. We believe that stewardship starts from within so that we can bring focus, creativity, and excellence to our work. Each of us is personally responsible for fostering a workplace where people can thrive. And we are stewards of the greater good of the company. By holding ourselves and each other accountable, we build a culture of innovation and collective success that reflects the scale of our mission. • Stewardship: • Courage: Courage is what unlocks the seemingly impossible for our customers. It’s the core value that drives us make bold moves and take on big, complicated network problems—the ones others avoid. We know success isn't guaranteed, but we have the audacious vision to believe a solution is possible and to build it. Courage fuels our growth mindset. It means embracing challenges that make us stronger, and it’s demonstrated by how we approach hard conversations and complex projects. • Courage: