Senior Machine Learning Engineer at Axon | San Francisco

The Senior Machine Learning Engineer at Scale AI is tasked with harnessing advanced methodologies in generative AI, computer vision, reinforcement learning, and agentic AI to enhance our product offerings and streamline the customer experience in live environments. Our machine learning engineers utilize a robust internal infrastructure and unparalleled access to extensive datasets to drive significant improvements for our clients. As part of the Public Sector Machine Learning team, you will be at the forefront of deploying innovative models to critical government systems through groundbreaking products such as Donovan and Thunderforge. Our work encompasses various modalities, with a strong emphasis on large language models and computer vision. On the LLM front, we are developing sophisticated agentic systems designed to address intricate operational and planning challenges for government entities. This involves creating agent frameworks that interface with custom retrieval systems and production APIs, alongside evaluation tools for assessing and refining agent performance. Additionally, we are pioneering research in reinforcement learning for agentic LLMs, successfully deploying these innovations in real-world operational contexts. In the realm of computer vision, we are focused on training advanced models to boost labeling throughput and automate perception tasks. Our initiatives include developing large-scale fine-tuning pipelines, training models across diverse modalities, and creating adaptable vision foundation models to support a variety of defense applications.

Senior Machine Learning Engineer - Model Evaluations for the Public Sector The Public Sector Machine Learning team at Scale AI pioneers the deployment of cutting-edge AI systems, including Large Language Models (LLMs), agentic models, and comprehensive multimodal pipelines, within critical government operations. We establish robust evaluation frameworks that ensure these models function reliably, safely, and effectively in real-world scenarios. As a Senior Machine Learning Engineer, you will architect, implement, and enhance automated evaluation pipelines that empower our clients to trust and effectively utilize advanced AI systems in defense, intelligence, and federal missions. Your Responsibilities Include: Creating and maintaining automated evaluation pipelines for machine learning models, focusing on functional, performance, robustness, and safety metrics, including evaluations based on LLM judges. Designing test datasets and benchmarks to assess generalization, bias, explainability, and potential failure modes. Building evaluation frameworks for LLM agents, which includes the infrastructure for scenario-based and environment-based testing. Conducting comparative analyses of model architectures, training procedures, and evaluation results. Implementing tools for continuous monitoring, regression testing, and quality assurance of machine learning systems. Designing and executing stress tests and red-teaming workflows to identify vulnerabilities and edge cases. Collaborating with operations teams and subject matter experts to generate high-quality evaluation datasets. This position requires an active security clearance or the ability to obtain one.

About Ditto Ditto builds technology that keeps data moving reliably, even when networks are unreliable or unavailable. Developers use Ditto’s peer-to-peer synchronization engine to power real-time applications for places where connectivity is a challenge: stadiums, aircraft, and remote military bases. Trusted by organizations such as Chick-fil-A, Delta Airlines, and the U.S. military, Ditto supports mission-critical operations across aviation, retail, travel, hospitality, and defense. The company has raised over $145 million and continues to grow globally, with a strong commitment to diversity and inclusion. Role Overview: Senior Software Engineer – Public Sector This Senior Software Engineer (Forward Deployed Engineer) position focuses on supporting Ditto’s public sector users. The role centers on collaborating directly with key users, integrating Ditto’s technology into operational environments, and ensuring rapid, effective deployments. The engineer acts as a technical leader in the field, solving complex problems as they arise and serving as a bridge between users and Ditto’s core engineering team. Adaptability and comfort with ambiguity are essential, as is the ability to resolve issues quickly in high-stakes situations. What You Will Do ATAK Plugin Development: Lead the design, creation, and field deployment of custom ATAK (Android Team Awareness Kit) plugins. These plugins embed Ditto’s data synchronization into the primary situational awareness tools used by warfighters. Android-Native Integrations: Build and optimize Android applications and services for tactical end-user devices. Ensure these solutions work reliably in disconnected, intermittent, and low-bandwidth (DIL) conditions. Integration Architecture: Design the full integration between Ditto’s platform and the broader TAK ecosystem. Define data contracts, API boundaries, and synchronization patterns to enable smooth interoperability with third-party systems and services. Real-Time Issue Resolution: Serve as the first line of technical support, diagnosing and fixing application, integration, and data-flow issues as they happen in the field. Location Remote, with preference for candidates based in Atlanta, Austin, San Francisco, or Seattle.

Join our innovative Match Team as a Senior Machine Learning Engineer at Enigmaio, where you will play a pivotal role in developing advanced algorithms and models that enhance our matching capabilities. You will collaborate with cross-functional teams to design, implement, and optimize machine learning solutions that drive business outcomes.

As a Senior Software Engineer within the Public Sector team at Scale AI, you will play a pivotal role in creating foundational products that empower forward-deployed teams to develop advanced agentic capabilities across various domains. Your responsibilities will include designing and building robust systems to ingest and process federal datasets, facilitating real-time decision-making in challenging environments. Our innovative solutions encompass:Establishing multi-layered guardrails around autonomous agentsEnhancing data retrieval processes for agentsCoordinating fleets of asynchronous agentsAutomatically notifying users of data anomaliesProviding transparency into how agents reach their conclusionsIn this role, you will lead the development of either a vertical feature or a horizontal capability, from defining requirements in collaboration with stakeholders through to final implementation and acceptance.Your core responsibilities include:Designing and implementing scalable backend systems and distributed architectures tailored for Federal clients.Managing the entire feature development lifecycle, from requirement gathering to deployment on classified networks.Overseeing the orchestration of asynchronous agent fleets to fulfill mission objectives.Engaging directly with customers to translate their mission needs into actionable technical requirements.Maintaining communication with stakeholders to ensure that implementations meet established acceptance criteria.Conducting technical reviews and assessing potential risks within machine learning infrastructure and model serving.Influencing the platform roadmap by delivering technical specifications for Federal product solutions.

About Taskrabbit:Taskrabbit is an innovative marketplace platform that seamlessly connects individuals with Taskers to manage everyday home tasks, including furniture assembly, handyman services, moving assistance, and much more.At Taskrabbit, we aim to transform lives one task at a time. We celebrate innovation, inclusion, and hard work, fostering a collaborative, pragmatic, and fast-paced culture. We seek talented, entrepreneurially minded, data-driven individuals who possess a passion for empowering others to pursue their passions. In partnership with IKEA, we are creating more opportunities for individuals to earn a consistent, meaningful income on their terms by establishing enduring relationships with clients in communities globally.Taskrabbit operates as a hybrid company, with team members located across the US and EU, and has been recognized as a Built In — Best Places to Work for 2022, 2023, and 2024, receiving accolades across various national and regional categories. Join us at Taskrabbit, where your contributions will be significant, your ideas appreciated, and your potential maximized!This position operates on a hybrid schedule, requiring two days of in-office collaboration per week. It can be based in our San Francisco office or our new New York City office (opening March 2026).About the RoleMachine Learning is a foundational element at Taskrabbit, and we are in search of an experienced Senior Machine Learning Engineer to join our team and help mold the future of ML/AI at Taskrabbit. This distinct, full-stack role is designed for someone who is enthusiastic about the entire machine learning lifecycle—from initial research and model development to constructing the robust infrastructure necessary for deploying and scaling your innovations.As a Senior Machine Learning Engineer, you will engage with exciting challenges that directly influence how users discover and interact with home services on the Taskrabbit platform. You will play a vital role in enhancing our capabilities in areas such as search ranking, content discovery, and recommendation systems. Collaborating closely with data scientists and fellow engineers, you will design and implement cutting-edge algorithms, ensuring the scalability, reliability, and optimization of our models in production alongside software engineers.

About VSCO At VSCO, we empower photographers with an innovative platform that provides essential tools, a vibrant community, and the visibility needed for creative and professional growth. We cultivate an authentic creative environment that welcomes photographers of all skill levels, offering a space that inspires opportunity, collaboration, and connection. Our mission is to support photographers in their journeys, enabling them to thrive and connect with fellow creatives and businesses through our comprehensive suite of tools, available on both mobile and desktop. We seek individuals who are passionate and proactive in advancing our mission. Our team members have the opportunity to make a significant impact, and we believe that collaborative efforts yield stronger results. Our core values are essential to our team culture and guide our hiring process. Learn more about what you can expect when joining VSCO on our Careers Page. About The Role As a Senior Machine Learning Engineer, you will harness the power of AI and machine learning to create innovative, reliable user-facing product features. You will leverage your extensive technical background and hands-on experience in deploying machine learning models to deliver impactful solutions based on real-world feedback. Your focus on measurable outcomes and customer satisfaction drives your work, blending innovation with practical implementation. You will be highly skilled in Python and adept across the data and machine learning stack, enabling you to develop and launch models efficiently while ensuring scalability and maintainability. Whether working with traditional algorithms or cutting-edge deep learning and generative AI, you will expertly navigate the complexity of each problem, managing every phase from defining the challenge to deployment and iterative improvement. Your dedication to software engineering excellence will inform your thoughtful approach to system design for machine learning, encompassing data quality, pipeline design, feature workflows, model serving, and ongoing monitoring and enhancement. By integrating machine learning deeply within our cohesive product experiences, you will collaborate effectively with cross-functional teams, aligning on objectives, defining success metrics, and driving meaningful outcomes. You will stay informed about the rapidly evolving AI landscape, maintaining a discerning perspective that allows your team to focus on significant advancements while avoiding distractions. The Day to Day Design and implement ML-powered features for search, discovery, personalization, and more.

Join Orchard as a Machine Learning Engineer and play a pivotal role in transforming data into actionable insights. In this dynamic position, you will leverage your expertise in machine learning algorithms and data analysis to develop innovative solutions that enhance our products and services.We are looking for a proactive team player who thrives in a fast-paced environment and possesses strong problem-solving skills. You will collaborate with cross-functional teams, engage with large datasets, and contribute to the design and implementation of machine learning models.

Join Cloudflare as a Senior Named Solutions Engineer specializing in the Public Sector. In this pivotal role, you will leverage your technical prowess and strong communication skills to deliver tailored solutions that meet the unique needs of governmental and public institutions. You will work collaboratively with various teams to drive successful outcomes and ensure that our clients receive the highest level of service and support.Your responsibilities will include engaging with stakeholders to understand their requirements, designing and implementing solutions, and providing ongoing technical guidance. You will also play a key role in educating clients about our products and services, ensuring they can maximize their use of our cutting-edge technology.

Are you ready to push the boundaries of what you believe you're capable of? At Coinbase, our vision is to enhance economic freedom globally. This is a grand, ambitious endeavor that challenges us to deliver our best every day as we construct the foundational onchain platform and shape the future of the global financial system.To drive our mission forward, we are in search of a unique candidate. We seek an individual who is not only passionate about our objective but also believes in the transformative power of cryptocurrency and blockchain technology to revolutionize the financial landscape. We are looking for someone eager to make a significant impact, who thrives under pressure while collaborating with a team of highly skilled professionals, and who actively seeks constructive feedback for continuous improvement. We want a problem-solver who embraces challenges head-on.Our work culture is intense and not suited for everyone. However, if you aspire to build the future alongside exceptional individuals and are ready to meet high expectations, this is the place for you.While many positions at Coinbase are remote-first, we are not solely remote. In-person engagements are expected throughout the year. We conduct team and company-wide offsites several times a year to promote collaboration, connection, and alignment. Your attendance is both expected and fully supported.We are looking for a Senior Machine Learning Platform Engineer to join our Machine Learning Platform team. This team is responsible for developing the core components for feature engineering, as well as training and serving ML models at Coinbase. Our platform plays a crucial role in combating fraud, personalizing user experiences, and analyzing blockchains. You will have the opportunity to leverage your engineering expertise across various aspects of large-scale ML development, including stream processing, distributed training, and highly available online services.

Join Arcade as a Senior Machine Learning Engineer, where you'll be at the forefront of AI innovation. In this pivotal role, you will leverage advanced machine learning algorithms to create and enhance cutting-edge solutions. Collaborate closely with cross-functional teams to drive impactful projects from ideation to deployment.

Saris AI develops applied AI solutions for the banking sector, with teams in San Francisco, Montreal, and Toronto. The company builds automation tools that handle complex, long-context reasoning and agent-driven decision-making. Reliability and compliance shape every product, and Saris AI's agents already manage real customer workflows in production. As revenue grows, the engineering team is expanding to enhance current offerings and explore new directions. The Senior Machine Learning Engineer role is based in San Francisco and sits within the core engineering group. The team works in a collaborative, early-stage setting, balancing infrastructure needs with the delivery of features that serve customers directly. What you will do Build and maintain machine learning infrastructure, such as evaluation frameworks, prompt management systems, and tools for model observability. Develop new AI features for customers while supporting and improving the underlying infrastructure. Shape strategies for evaluation, LLM routing, prompt engineering, and model selection. Set practical standards to boost quality without slowing down development. Guide technical direction by clarifying trade-offs and architectural choices. Requirements Minimum 4 years of experience in machine learning or AI engineering, including production deployment of ML systems. Direct experience with large language models, prompt engineering, evaluation techniques, and model routing. Background in building tools and systems that deliver value to users. Comfort making pragmatic trade-offs and recognizing when a solution is sufficient. Ability to navigate ambiguity, define problems, and deliver results independently. Strong focus on end users and understanding the impact of ML decisions on customer experience. Supports team growth through code reviews, collaboration, and clear technical communication. Bonus Experience in regulated industries, especially banking.

Highlight is pioneering a shared intelligence framework for the contemporary workforce. Our solution integrates context across every member and tool in your team, effectively eliminating information silos. As your organization evolves, Highlight adapts by intelligently routing knowledge and reliably automating workflows.The RoleWe are seeking a Senior Machine Learning Engineer to contribute to the development of the AI systems that drive Highlight's operations. This position involves working across the entire ML stack, including data pipelines, model training, retrieval, ranking, evaluations, and deployment. You will deliver features that create a seamless experience for our users. This is a hands-on individual contributor role where you will manage critical systems from inception to deployment, collaborating closely with our Head of AI Engineering to enhance our ML capabilities.You will excel in this role if you are passionate about delivering production-ready ML systems, maintaining high standards of output quality, and tackling complex problems that have a tangible impact on users. We prioritize speed, uphold rigorous standards, and believe in the merit of ideas, regardless of hierarchy.Note: This is an on-site position, requiring five days a week at our San Francisco or New York City offices.ResponsibilitiesDesign, develop, and refine ML pipelines for context retrieval, action detection, and output generation.Implement and iterate on Retrieval-Augmented Generation (RAG) systems, ranking models, and prompt engineering to enhance quality.Establish comprehensive evaluations and monitoring frameworks to assess and enhance ML system performance.Explore alternative models and fine-tuning strategies for our substantial background processing workloads, creating improved systems that enhance quality while optimizing cost efficiency.Work in collaboration with Backend and Product Engineering teams to deliver AI-powered features.Influence ML infrastructure decisions and establish best practices across the engineering organization.Stay updated on advancements in LLMs, retrieval techniques, and ML tools; share insights with the team.Candidate ProfileWe are looking for an individual who is excited about crafting the AI user experience for the future. We value extreme ownership, accountability, and proactivity as we strive to exceed expectations. The ideal candidate is a skilled craftsman who enjoys developing exceptional products utilizing LLMs at scale.You are a suitable candidate if you possess:4+ years of experience in ML/AI engineering, with substantial involvement in deploying production systems.

Join Cloudflare as a Senior Solutions Engineer focusing on the Public Sector Federal team. In this pivotal role, you will harness your expertise in solutions engineering to deliver innovative cloud solutions to federal clients. You will collaborate closely with cross-functional teams to understand client needs and craft tailored solutions that address their unique challenges.Your primary responsibilities include engaging with customers to gather requirements, presenting technical solutions, and ensuring successful implementations, while also providing ongoing support to maximize customer satisfaction.

Join Our Team at MacroscopeAt Macroscope, we are dedicated to being the definitive source of truth for any software development company. Our mission is to empower leaders with clarity and provide engineers with the time they need to innovate.We enable leaders to gain insights into the evolution of their products and codebases—tracking changes, understanding team contributions, and identifying progress—all grounded in the ultimate source of truth: the code itself.Founded by experienced entrepreneurs who have successfully built and sold multiple companies, and held executive positions in public tech firms, we are backed by top-tier venture capital firms such as Lightspeed Venture Partners, Thrive Capital, Google Ventures, and Adverb.The RoleWe are seeking a Senior Applied Machine Learning Engineer who will be responsible for designing, developing, and optimizing the ML and AI systems that drive our core offerings. You will have full ownership of the systems, overseeing everything from data collection and evaluation to model experimentation and large-scale production deployment.This cross-functional position entails leading the ML/AI lifecycle for one of our most vital features: AI Code Review. Collaborating closely with our co-founders, you will make pivotal decisions that shape our product's development—ranging from building high-quality datasets to interpreting experimental results and enhancing model performance architecture. Additionally, you will play a significant role in crafting and implementing software that seamlessly integrates our models with our backend applications and user experience, offering a unique opportunity to influence our product's evolution significantly.Technology Stack: Typescript/React (frontend), Golang (backend), Temporal, Google Cloud (GCP), Postgres, Terraform, and custom-built AST "code walkers" in several programming languages including Golang, Typescript, Swift, Python, and Rust.

Join Handshake as a Machine Learning Engineer I, where you will have the opportunity to work on cutting-edge machine learning projects that drive our innovative solutions. Collaborate with a talented team to develop algorithms and models that enhance our product offerings and improve user experiences.

Join Our Innovative Team at HiveHive is at the forefront of cloud-based AI solutions, revolutionizing how organizations understand, search for, and generate content. Trusted by many of the world's largest and most groundbreaking companies, we empower developers with premier pre-trained AI models that handle billions of API requests monthly. Our turnkey software applications leverage proprietary AI models and datasets, driving transformative advancements in content moderation, brand protection, sponsorship measurement, and context-based ad targeting.With over $120M in funding from prominent investors like General Catalyst, 8VC, Glynn Capital, Bain & Company, and Visa Ventures, Hive is rapidly expanding. Our dynamic team of over 250 employees operates from our San Francisco, Seattle, and Delhi offices. If you are passionate about shaping the future of AI, we invite you to explore opportunities with us!About the Machine Learning Engineer RoleAs we strive to achieve our ambitious vision, we seek exceptional machine learning engineers to join our team. We are looking for enthusiastic developers who are eager to remain at the cutting edge of deep learning technology, designing and deploying state-of-the-art neural network models into production. Our ideal candidates thrive in working with large-scale datasets and demonstrate a keen interest in mastering new technologies across the machine learning spectrum. We value individuals who are proactive and take ownership of their projects, contributing innovative ideas and practical implementations. Experience in building machine learning applications from the ground up and designing scalable, maintainable data pipelines is essential.

Join our dynamic Personalization team at Boomtrain as a Machine Learning Engineer. We are in search of a skilled engineer who will play a pivotal role in developing and enhancing our recommendation systems that cater to a variety of customers.In this role, you will collaborate with a talented team dedicated to designing and implementing innovative models and systems that deliver personalized recommendations. You will have the opportunity to work on complex engineering challenges and contribute to generating hundreds of millions of recommendations daily.This position offers a unique chance to engage in end-to-end project work and make a significant impact on our personalization initiatives.Key Responsibilities:Research and propose advanced recommendation and optimization models to enhance our personalization systems.Develop and maintain offline model generation pipelines.Design and maintain online recommendation serving systems.

About FaireFaire is a transformative online wholesale marketplace, driven by the conviction that local businesses are the future. Independent retailers around the globe generate more revenue than massive corporations like Walmart and Amazon combined, yet individually, they remain small. At Faire, we harness technology, data, and machine learning to connect this vibrant community of entrepreneurs. Think of your favorite local boutique — we empower them to discover and sell the best products from around the world. With our innovative tools and insights, we aim to level the playing field, enabling small businesses to thrive against larger competitors.By championing the growth of independent businesses, Faire positively impacts local economies on a global scale. We’re in search of intelligent, resourceful, and passionate individuals to join us in fueling the shop local movement. If you value community, we invite you to be part of ours.About this RoleAs the Senior Staff Machine Learning Platform Engineer, you will spearhead the technical vision and evolution of Faire's ML platform. You will establish standards, influence organization-wide architecture, and lead intricate, cross-functional initiatives that enhance data science velocity at scale. This position is crucial for adapting ML workflows to leverage modern AI productivity tools. You will not only develop models but also design the systems that enable those models to empower tens of thousands of small retailers in competing and growing their local businesses.