GrepJob
Axon

Machine Learning Engineer II

Axon
Apply
about 5 hours ago
Ho Chi Minh City, VietnamMid Level / Senior
H1B Sponsor

Responsibilities

  • Collaborate with scientists and product managers to build proof-of-concepts.
  • Architect and develop secure, privacy-preserving AI solutions.
  • Create the next generation of continuous deployment frameworks for AI.
  • Optimize AI models for deployment on existing and new Axon devices.
  • Work with scientists to implement state-of-the-art training techniques.

Requirements

  • Bachelor’s Degree in a technical field such as Computer Science or Engineering.
  • 6+ years of software engineering experience with a focus on deploying AI models.
  • Experience with GPU/TPU development and debugging.
  • Proficiency in Python and C++, with familiarity in ML frameworks like TensorFlow or PyTorch.
  • Advanced knowledge of Linux and excellent problem-solving skills.

Benefits

  • Medical, Dental and Vision Insurance.
  • Robust Paid Time Off policy.
  • Bonuses and lunch allowance.
  • Cell phone stipend and free LinkedIn Learning or Udemy account.
  • Access to 24/7 online emotional and mental support.

Tech Stack

Categories

AI & MLData Science