Clash
of Jobs
Browse All Jobs
Job Description
Axon is seeking an AI Research Scientist II, LLM, to contribute to its mission of protecting life. This role involves addressing challenging problems in the public safety space through AI and Machine Learning. The Research Scientist will investigate research approaches, assess implementation risks, and define success metrics for AI/ML projects, collaborating with other scientists and engineers. They will be actively involved in the entire AI innovation lifecycle, from model prototyping to deployment and continuous learning.
Role Involves:
- Driving phases in ML development, including shaping datasets and investigating ML architectures.
- Training, evaluating, and tuning ML models.
- Implementing end-to-end pipelines.
- Leveraging state-of-the-art research to deliver high-quality models.
- Contributing to the research community through publications and open-source code.
Requirements:
- A Master’s Degree in Computer Science, Machine Learning, or a related field.
- 3+ years of combined academic and industrial research experience developing LLM and other NLU models.
- Experience in big data ML and data-efficient ML techniques.
- Hands-on experience with Python, PyTorch, and TensorFlow.
- Experience in prompt engineering and fine-tuning ML models.
- Familiarity with NLU/LLM cloud services and APIs.
- Deep understanding of metrics for offline and online evaluation of LLM-based systems.
- Excellent problem-solving and communication skills.
What Axon Offers:
- Competitive salary and benefits.
- Medical insurance plan.
- Free lunch, gym, phone stipend, and parking.
- Fun team activities and outings.
- Flexible working hours.
- Opportunities for training and rotations in the US.
- Opportunities to ride along with real US police officers.
