AI Research Scientist II, LLM - Remote

Overview

Join Axon and be a Force for Good. At Axon, we’re on a mission to Protect Life. We’re explorers, pursuing society’s most critical safety and justice issues with our ecosystem of devices and cloud software. Like our products, we work better together. We connect with candor and care, seeking out diverse perspectives from our customers, communities and each other.

In Short

• Drive one or more phases in ML development: shape datasets, investigate ML architectures, train/evaluate/tune ML models, implement end-end pipeline.
• Leverage state-of-the-art research to deliver high quality models enabling multiple AI projects at scale.
• Contribute back to the research community via academic publications, tech blogs, open-source code and contributing to internal/external AI challenges.

Requirements

• A Master’s Degree in Computer Science, Machine Learning, Statistics, Applied Mathematics or an equivalent highly technical field.
• 3+ years of combined academic and industrial research experience developing LLM and other NLU models.
• Hands on experience in developing, scaling and implementing machine learning solutions using relevant programming languages (such as Python), state-of-the-art deep learning frameworks (such as PyTorch and Tensorflow).
• Experience in big data ML as well as data efficient ML that leverages techniques such as synthetic data construction, transfer learning, active learning, semi-supervised learning, few-shot learning.
• Experience in developing LLM-based applications including agent-based systems, RAG-based systems.
• Deep understanding of metrics for offline and online evaluation of LLM-based systems.
• Excellent problem solving skills and ability to dive into learning optimization, model architecture, evaluation metrics, and field testing scenarios.

Benefits

• An awesome office/working environment with the latest technologies.
• Competitive salary and benefits including a great medical insurance plan for you and your family.
• Free lunch, gym, phone stipend, parking.
• Flexible working hours.
• Opportunities for training and rotations in the US.