Axiado

Staff Machine Learning Engineer

Axiado
Apply
3 months ago
Taipei, TaiwanStaff+

Responsibilities

  • Act as a senior individual contributor and technical architect through hands-on coding, design, analysis, and technical leadership across the ML stack.
  • Define end-to-end architecture for MLOps, agentic AI, model optimization, and security strategies.
  • Design data processing, data versioning, labeling, active-learning, and production monitoring pipelines.
  • Build human-in-the-loop and AI-in-the-loop systems to continuously improve datasets.
  • Develop lightweight on-device monitoring for operational metrics, inference quality, and concept drift.
  • Design and develop autonomous agents for resource-constrained edge devices using log analysis, computer vision, and open-source system tools.
  • Implement model security and verification capabilities including secure protocols, authentication, model signing, and model injection prevention.
  • Bridge ML software development with silicon teams and collaborate with RTL designers on NPU and FPGA architecture.
  • Lead model optimization research using graph-level techniques such as operator fusion and operator-level techniques such as custom operations.

Requirements

  • 8–10+ years of hands-on machine learning experience with a senior- or staff-level individual contributor track record.
  • Ph.D. or M.S. in Computer Science, Electrical Engineering, or a related field, or equivalent practical experience.
  • Expert-level Python programming and deep experience with ML frameworks such as PyTorch and TensorFlow.
  • Deep theoretical understanding of modern machine learning algorithms such as Transformers.
  • Foundational understanding of computer architecture, digital logic, and RTL, including Verilog/VHDL's role in hardware design.
  • Experience architecting and building complete MLOps lifecycles from data ingestion through production monitoring and labeling loops.
  • Experience developing agentic systems or applications using LLMs.
  • Domain knowledge in log analysis and/or computer vision.
  • Experience with on-device model security, verification, anti-injection techniques, and secure communication protocols.
  • Hands-on experience optimizing models for NPUs or GPUs at graph and operator levels.
  • Preferred experience with ML compilers such as Apache TVM or MLIR.
  • Preferred hands-on experience with Kubernetes for MLOps, including Kubeflow or Argo.
  • Preferred familiarity with GPU scheduling and virtualization platforms such as Run:AI.
  • Preferred embedded systems experience with C++, Rust, or Yocto.
  • Preferred familiarity with the RISC-V instruction set architecture.
  • Preferred proficiency with AWS, GCP, or Azure.

Benefits

  • The company describes a diverse, collaborative, continuous-learning environment with mutual support.
  • The role is based at the company headquarters in Silicon Valley; no remote or hybrid arrangement is stated.
  • Axiado is an Equal Opportunity Employer.
Axiado

About Axiado

51-200 employees
Contact me