Machine Learning Engineer II
| Company | Axon |
|---|---|
| Location | San Francisco, CA, USA |
| Salary | $140000 – $210000 |
| Type | Full-Time |
| Degrees | Bachelor’s, Master’s, PhD |
| Experience Level | Senior |
Requirements
- Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field.
- 5+ years of software engineering experience and a proven track record of successfully architecting and maintaining large-scale distributed platforms for AI.
- Proficiency in python, familiarity with ML frameworks such as TensorFlow, Keras, PyTorch.
- Advanced knowledge and hands-on experience with at least one cloud environment (such as AWS, Microsoft Azure Oracle, or Google Cloud).
- Excellent problem solving and software design skills.
- Understanding of system architecture.
- Comfort communicating and interacting with scientists, engineers and ML/product managers.
Responsibilities
- Architect, develop and manage a scalable, reliable, optimized, infrastructure to train, test, and improve Automatic Speech Recognition, Computer Vision and Natural Language Understanding models (including LLMs).
- Deliver novel, strategic solutions to improve the fairness of Axon models.
- Deliver novel, strategic solutions to improve the security of Axon models.
- Deliver novel, strategic solutions to accelerate innovation.
- Influence to the AI community by developing state-of-the-art solutions for responsible AI.
Preferred Qualifications
- Master’s Degree/PhD in Computer Science, Engineering, Electronics, Mathematics or an equivalent highly technical field.
- Hands-on experience in solving Computer Vision or Speech Recognition problems in a business setting.
- Familiarity with responsible AI, de-biasing, encryption and de-identifying techniques.