Axon

Senior Machine Learning Engineer

Axon

Apply
about 1 month ago
San Francisco, CA, USA
Senior / Mid Level
H1B Sponsor

Base Salary

$173k - $277k/yr

Responsibilities

  • Collaborate with scientists and product managers to build proof-of-concepts.
  • Architect and develop secure, privacy-preserving AI solutions.
  • Architect platforms that accelerate research and AI product development.
  • Implement state-of-the-art training techniques in collaboration with scientists.
  • Set high standards for ethical and responsible AI development.

Requirements

  • Bachelor’s Degree in a technical field such as Computer Science or Engineering.
  • 6+ years of software engineering experience with a track record of deploying AI models.
  • Experience with Infrastructure-as-code and cloud architecture.
  • Proficiency in Python and C++, with familiarity in ML frameworks like TensorFlow or PyTorch.
  • Advanced knowledge and hands-on experience with Linux.
  • Excellent problem-solving and software design skills.
  • Strong communication skills to interact with diverse teams.

Benefits

  • Competitive salary and 401k with employer match.
  • Discretionary paid time off.
  • Paid parental leave for all employees.
  • Medical, Dental, and Vision plans.
  • Fitness programs and emotional wellness support.
  • Learning and development programs.
  • Snacks available in the office.

Tech Stack

C++LinuxPythonPyTorchTensorFlow

Categories

AI & MLData Science