← All Positions
Posted May 21, 2026

**Senior Customer Success Machine Learning Engineer – Public Sector Practice**

Apply Now
At arenaflex, our mission is to empower AI developers with the best tools for building better models. As a pioneer in the AI developer platform space, we've expanded our solution to cater to organizations focused on building their own deep learning models and generative AI applications. With a strong foundation of over $250M in funding and a team of over 200 employees, we proudly serve over 1,000 customers and more than 30 foundation model builders, including industry leaders like OpenAI, NVIDIA, Microsoft, and Toyota. We're seeking a highly skilled Senior Customer Success Machine Learning Engineer to lead the technical engagement and success of our federal and public sector clients. Reporting to the Head of Field Engineering and partnering closely with our Head of Federal, this role will be highly visible and critical to the growth of arenaflex's burgeoning public sector practice. In this role, you will lead pre- and post-sales motions, working closely with federal machine learning (ML) teams to tackle critical, real-world challenges across computer vision, NLP, robotics, and help government clients understand how arenaflex can help with evolving AI assurance requirements. **Key Responsibilities:** * Lead pre- and post-sales motions, guiding federal ML teams from initial demo to full deployment and customer success * Serve as a government vertical's technical GTM representative for government needs internally across Product and Engineering teams * Coordinate the support of post-sale technical and implementation requirements aligned with government standards * Collaborate with federal ML teams to tackle critical, real-world challenges across computer vision, NLP, robotics, and help government clients understand how arenaflex can help with evolving AI assurance requirements * Develop and maintain strong relationships with federal ML teams, understanding their needs and pain points to provide tailored solutions * Stay up-to-date with the latest advancements in ML and AI, ensuring arenaflex's platform remains competitive and effective in meeting federal compliance standards * Work closely with the Head of Federal to develop and execute a comprehensive strategy for arenaflex's public sector practice **Essential Qualifications:** * Up to 8 years of relevant experience in a similar role working with the public sector or with government clients * Based in the D.C. metro area * Experience using one or more of the following packages: TensorFlow/Keras, PyTorch Lightning * Strong programming proficiency in Python and willingness to work "hands-on keyboard" to help practitioners see the value of arenaflex's platform * Ability to adapt machine learning solutions to meet federal compliance standards and an understanding of what it takes to get the technical win in the government market * Excellent communication and presentation skills, both written and verbal, with a focus on explaining technical concepts within the context of federal guidelines **Preferred Qualifications:** * Experience with cloud platforms (AWS, GCP, Azure), preferably in environments compliant with FedRAMP or government standards * Experience with Linux/Unix * Familiarity with arenaflex's platform and its capabilities * Experience working with government clients and understanding their unique needs and challenges **Skills and Competencies:** * Strong technical expertise in machine learning and AI, with the ability to adapt solutions to meet federal compliance standards * Excellent communication and presentation skills, with the ability to explain technical concepts in a clear and concise manner * Strong relationship-building skills, with the ability to develop and maintain strong relationships with federal ML teams * Ability to work effectively in a fast-paced environment, managing multiple conflicting priorities and responding promptly to changing needs * Strong problem-solving skills, with the ability to break down complex problems and resolve them through customer consultation and execution **Career Growth Opportunities and Learning Benefits:** * arenaflex is committed to providing opportunities for growth and development, with a focus on helping employees achieve their career goals * Collaborate with a talented team of engineers, researchers, and product managers to develop and refine arenaflex's platform * Stay up-to-date with the latest advancements in ML and AI, with access to training and development opportunities to enhance your skills and knowledge * Participate in arenaflex's mentorship program, pairing you with experienced engineers and researchers to provide guidance and support **Work Environment and Company Culture:** * arenaflex is a remote-first company, with a flexible work environment that allows you to work from anywhere * Collaborate with a talented team of engineers, researchers, and product managers to develop and refine arenaflex's platform * Participate in arenaflex's social events and activities, including team-building exercises, hackathons, and community outreach programs * Enjoy a comprehensive benefits package, including flexible time off, medical, dental, and vision coverage, and a home office budget with a new high-powered laptop **Compensation, Perks, and Benefits:** * Truly competitive salary and equity * Flexible time off * Medical, dental, and vision coverage for employees and family members * Remote first culture with in-office flexibility in San Francisco * Home office budget with a new high-powered laptop * 12 weeks of parental leave (U.S. specific) * 401(k) (U.S. specific) * Supplemental benefits may be available depending on your location **How to Apply:** If you're a highly skilled machine learning engineer with a passion for working with government clients, we encourage you to apply for this exciting opportunity. Please submit your resume and a cover letter explaining why you're the ideal candidate for this role. We can't wait to hear from you! Apply for this job