Staff Software Engineer - AI

onX • Bozeman, MT

5h • Remote

About The Position

onX is seeking a Staff Software Engineer who will contribute to meaningful, real-world outcomes that have a lasting impact for customers and the business. In this role, you’ll focus on the design, development, and scaling of our core AI/ML platforms and product-facing services as part of the AI Foundations Team, supporting the integration of cutting-edge artificial intelligence solutions across onX's product lines. You’ll take ownership of your work, collaborate closely with partners across the organization, and help move initiatives forward in an environment where priorities evolve, learning is celebrated, and you’re encouraged to keep stretching your skills. This role will report to the Director of Engineering.

Requirements

• Bachelor’s degree in Computer Science, Data Science, or a related technical field, or equivalent experience.
• 10+ years of professional software engineering experience, with at least 3 years focused on building and scaling production AI/ML systems, is required.
• Expertise in building scalable backend systems.
• Deep knowledge of MLOps practices, cloud infrastructure (AWS/GCP/Azure), and CI/CD pipelines for machine learning models and the common inference APIs/SDKs and frameworks.
• Experience with modern AI technologies, including Large Language Models (LLMs) and their various architectures, methods such as RAG, vector databases, and the common AI stack including TensorFlow/Keras, PyTorch, Transformers, etc.
• A strong curiosity for exploring new technologies, including AI
• A shared passion for and ability to demonstrate onX’s Company Values.
• Permanent US work authorization is a condition of employment with onX.

Nice To Haves

• A history of engaging in AI development, from the early ML/NN (YOLO, Darknet) to the LLM era.
• Experience in monitoring and debugging production AI systems (inference, evals, judging, benchmarks) including cost and performance optimizations.
• Interest or experience in edge/on-device inference (MLX, JAX, transformers on TPU/NPU, etc.)
• Interest, experience or experimentation in training/fine-tuning transformer/diffusion models, dataset curation and captioning, etc.

Responsibilities

• Work on the design, development, and deployment of large-scale AI systems and services, including models, pipelines, product experiences and infrastructure.
• Help Drive the technical strategy and roadmap for AI feature development, focusing on scalability, repeatability, reliability, and performance.
• Mentor and guide other engineers in best practices and productionizing AI technologies.
• Collaborate with Product and Data Science teams to translate business problems into technical requirements and deliver high-impact AI solutions (combining LLMs, Vector Databases, ML and other technology in novel ways to serve onX' unique audience).
• Look for opportunities to embed AI as a repeatable mechanism in daily workflows by integrating experimentation into real work, and continuously refining its use with sound judgment and validation.
• Be a leader in AI safety, ensuring AI-enabled products are aligned with our values and mission.
• Look for opportunities to embed AI as a repeatable co-pilot in daily workflows by integrating experimentation into real work, and continuously refining its use with sound judgment and validation.

Benefits

• Competitive salaries, annual bonuses, equity, and opportunities for growth
• Comprehensive health benefits, including a no-monthly-cost medical plan
• Paid parental leave of 13 weeks for birthing parents and 5 weeks for non-birthing parents
• 401k matching at 100% for the first 3% you save and 50% from 3-5%
• Company-wide outdoor adventures and amazing outdoor industry perks
• Annual “Get Out, Get Active” funds to fuel your active lifestyle
• Flexible time away package that includes PTO, STO, VTO, and paid holidays

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Number of Employees: 101-250 employees