Talent.com
Staff Machine Learning Engineer

Staff Machine Learning Engineer

AxiadoSan Jose, CA, United States
5 days ago
Job type
  • Full-time
Job description

Company Description

Axiado is an AI-enhanced security processor company redefining the control and management of every digital system. The company was founded in 2017, and currently has 150+ employees. At Axiado, developing great technology takes more than talent : it takes amazing people who understand collaboration, respect each other, and go the extra mile to achieve exceptional results. It takes people who have the passion and desire to disrupt the status quo, deliver innovation, and change the world. If you have this type of passion, we invite you to apply for this job.

Job Description

This is a "full-stack" ML systems role for a senior individual contributor and technical architect. You will be responsible for designing the complete ML ecosystem for our edge devices, from the cloud-native MLOps platform down to the bare-metal model optimization.

This unique role blends three key domains :

  • MLOps & Data : You will architect the entire data lifecycle, including our CI / CD pipelines, data-labeling loops, and on-device monitoring.
  • Agentic & Edge AI : You will lead the design of autonomous agents that run on our edge devices, using domain knowledge in log analysis and computer vision.
  • Systems & Hardware : You will be the "hardware-aware" expert, bridging our ML software with our silicon team to ensure our models are hyper-optimized for our custom NPU.

You are the engineer who will not only build our ML platform but also design the intelligent agents it deploys and ensure they run faster and more securely than anyone else's.

Key Responsibilities

Architecture & Leadership :

  • Act as a senior individual contributor, leading by example with hands-on coding, design, and analysis across the entire ML stack.
  • Define the end-to-end architecture for our MLOps, agentic AI, and model optimization strategy.
  • MLOps & Data Platform :

  • Design and implement our data processing and versioning pipelines, ensuring data integrity and traceability.
  • Build the infrastructure for our Human-in-the-Loop (HITL) and AI-in-the-Loop (Active Learning) data labeling systems to continuously improve our datasets.
  • Develop a comprehensive, lightweight on-device monitoring system to track not just operational metrics but also inference quality and concept drift.
  • Agentic & Edge Development :

  • Design and development of autonomous agents that operate on our resource-constrained edge devices.
  • Integrate deep domain knowledge, including real-time log analysis, computer vision, and interaction with open-source system tools.
  • Security & Optimization :

  • Define and implement the complete security and verification framework for our edge models. This includes MCP / A2A-like secure protocols, MCP authentication, entity verification (e.g., model signing), and model injection prevention.
  • Serve as the primary technical bridge to our silicon teams. Collaborate with RTL designers to influence future NPU and FPGA architecture from an ML software perspective.
  • Lead R&D on model optimization for our specific AI inference engine, applying both graph-level (e.g., operator fusion) and OP-level (e.g., custom ops) techniques.
  • Qualifications

  • 8-10+ years of hands-on experience in machine learning, with a proven track record as a senior or staff-level individual contributor.
  • Ph.D. or M.S. in Computer Science, Electrical Engineering, or a related field (or equivalent practical experience).
  • Expert-level programming in Python and deep experience with ML frameworks (e.g., PyTorch, TensorFlow).
  • Deep theoretical understanding of modern ML algorithms (e.g., Transformers).
  • A strong foundational understanding of computer architecture, digital logic, and the role of RTL (Verilog / VHDL) in the hardware design lifecycle.
  • Proven experience architecting and building end-to-end MLOps lifecycles, from data ingestion to production monitoring and labeling loops.
  • Proven experience developing agentic systems or applications using LLMs.
  • Demonstrable domain knowledge in log analysis AND / OR computer vision.
  • Experience with on-device model security (verification, anti-injection) and secure communication protocols.
  • Hands-on experience optimizing models for hardware (NPUs, GPUs) at graph and operator levels.
  • Preferred Qualifications (The "Plus" Factors)

  • Direct experience with ML compilers such as Apache TVM or MLIR.
  • Hands-on experience with Kubernetes (K8s) for MLOps (e.g., Kubeflow, Argo).
  • Familiarity with GPU scheduling and virtualization platforms like Run : AI.
  • Deep experience with embedded system development (C++, Rust, Yocto).
  • Familiarity with the RISC-V instruction set architecture (ISA).
  • Proficiency with cloud platforms (AWS, GCP, Azure).
  • Additional Information

    Axiado is committed to attracting, developing, and retaining the highest caliber talent in a diverse and multifaceted environment. We are headquartered in the heart of Silicon Valley, with access to the world's leading research, technology and talent.

    We are building an exceptional team to secure every node on the internet. For us, solving real-world problems takes precedence over purely theoretical problems. As a result, we prefer individuals with persistence, intelligence and high curiosity over pedigree alone. Working hard and smart, continuous learning and mutual support are all part of who we are.

    Axiado is an Equal Opportunity Employer. Axiado does not discriminate on the basis of race, religion, color, sex, gender identity, sexual orientation, age, non-disqualifying physical or mental disability, national origin, veteran status or any other basis covered by appropriate law. All employment is decided on the basis of qualifications, merit, and business need.

    Create a job alert for this search

    Staff Machine Learning Engineer • San Jose, CA, United States

    Related jobs
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    AarkiSan Francisco, CA, United States
    Full-time
    Aarki is an AI-driven company specializing in mobile advertising solutions designed to fuel revenue growth.We leverage AI to discover audiences in a privacy-first environment through trillions of c...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    QuantcastSan Francisco, CA, United States
    Full-time
    At Quantcast, we're redefining what's possible in digital advertising.As a global Demand Side Platform (DSP) powered by AI, we help marketers connect with the right audiences and deliver measurable...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    AdobeSan Jose, CA, United States
    Full-time
    Staff Machine Learning Engineer.Adobe Photoshop is seeking a Staff Machine Learning Services Engineer to serve as the technical lead for our Generative AI Services domain.In this high-impact role, ...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer, Level 6

    Staff Machine Learning Engineer, Level 6

    Snap Inc.Palo Alto, CA, United States
    Full-time
    Staff Machine Learning Engineer, Level 6 page is loaded.Staff Machine Learning Engineer, Level 6.Apply locations Palo Alto, California Seattle, Washington Santa Monica - 2850 Ocean Park Blvd San Fr...Show moreLast updated: 16 days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    HiveSan Francisco, CA, United States
    Full-time
    Hive is the leading provider of cloud-based AI solutions to understand, search, and generate content, and is trusted by hundreds of the world's largest and most innovative organizations.The company...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    SukiRedwood City, CA, United States
    Full-time
    Get AI-powered advice on this job and more exclusive features.The Future of Healthcare Needs You.At Suki, were building technology that listens, understands, and gets out of the way so clinicians c...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer, Applied Science

    Staff Machine Learning Engineer, Applied Science

    Hippocratic AIStanford, CA, United States
    Full-time
    Staff Machine Learning Engineers at Hippocratic AI are foundational to the design, deployment, and optimization of cutting-edge ML systems powering our next-generation, safety-focused generative AI...Show moreLast updated: 2 days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    SoFiSan Francisco, CA, United States
    Full-time
    Employee Applicant Privacy Notice.Shape a brighter financial future with us.Together with our members, we're changing the way people think about and interact with personal finance.We're a next-gene...Show moreLast updated: 2 days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    scribehow.comSan Francisco, CA, United States
    Full-time
    We're looking for a highly motivated and skilled Staff Machine Learning Engineer to build the future of AI productivity software. You will lead high-impact research and development, drive our core A...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer, Core Engineering

    Staff Machine Learning Engineer, Core Engineering

    PinterestSan Francisco, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show moreLast updated: 2 days ago
    • Promoted
    Sr. Staff Machine Learning Engineer, Homefeed

    Sr. Staff Machine Learning Engineer, Homefeed

    PinterestSan Francisco, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Moloco, Inc.Redwood City, CA, United States
    Full-time
    Moloco is a machine learning company empowering organizations of all sizes to grow and unlock the full value of their unique first-party data, elevating the traditional path to performance advertis...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning R&D Engineer

    Staff Machine Learning R&D Engineer

    Visual LeaseSunnyvale, CA, United States
    Full-time
    Matterport is leading the digital transformation of the built world.Our groundbreaking spatial computing platform turns buildings into data making every space more valuable and accessible.Millions ...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Lily AISan Francisco, CA, United States
    Full-time
    Staff Machine Learning Engineer.Staff Machine Learning Engineer.Lily AI is a female-founded retail AI company empowering retailers and brands by bridging the gap between merchant-speak and customer...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer, Acceleration

    Staff Machine Learning Engineer, Acceleration

    WaymoSan Francisco, CA, United States
    Full-time
    Staff Machine Learning Engineer, Optimization.Staff Machine Learning Engineer, Optimization.Staff Machine Learning Engineer, Optimization. Be among the first 25 applicants.Staff Machine Learning Eng...Show moreLast updated: 2 days ago
    • Promoted
    Staff Machine Learning R&D Engineer

    Staff Machine Learning R&D Engineer

    MatterportSunnyvale, CA, United States
    Full-time
    Matterport is leading the digital transformation of the built world.Our groundbreaking spatial computing platform turns buildings into data making every space more valuable and accessible.Millions ...Show moreLast updated: 6 days ago
    • Promoted
    Sr. Staff Machine Learning Engineer, Closeup Relevance

    Sr. Staff Machine Learning Engineer, Closeup Relevance

    PinterestPalo Alto, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show moreLast updated: 3 days ago
    • Promoted
    AIML - Staff Machine Learning Engineer

    AIML - Staff Machine Learning Engineer

    Apple Inc.Cupertino, CA, United States
    Full-time
    Cupertino, California, United States Machine Learning and AI.The Apple Knowledge & Information (AKI) Entity Resolution team is looking for senior and staff engineers to lead software projects suffu...Show moreLast updated: 30+ days ago