Principal Machine Learning Engineer

Axion Ray’s mission is to improve the quality and safety of innovative, engineered products - airplanes, electric vehicles, medical devices, home appliances, consumer electronics - by creating the world’s best proactive management platform, powered by the latest advances in artificial intelligence. Axion leverages bleeding-edge tech & AI stack to solve real-world problems. With investment from top Enterprise SaaS VCs like Bessemer, and key strategic partners such as Boeing and Raytheon, we are uniquely positioned to solve the hardest problems in manufacturing today. Our team includes experts in Enterprise AI from Palantir, McKinsey & Quantum Black, and other successful Enterprise SaaS companies. Since our founding at the onset of 2021, we’ve deployed across some of the largest manufacturers in the world. If you want the chance to help build the future of manufacturing, join us!

What you'll do:

• Design, implement, and scale critical machine learning components and services to support Axion’s most strategic initiatives

• Build a next-generation training framework that can support large-scale model training, enabling us to push the limits of what's possible with machine learning

• Perform training and model performance optimization with various GPUs to improve model training speed and efficiency

• Work across teams to understand product requirements, evaluate trade-offs, and deliver the solutions needed to build innovative products or services

• Advocate for and apply best practices when it comes to availability, scalability, operational excellence, and cost management

• Provide technical direction that influences the entire company

• Work in a hybrid environment based in NYC

Who you are:

• Strong understanding of machine learning approaches and algorithms

• Excellent programming and software design skills, including debugging, performance analysis, and test design

• Proven track record of operating highly-available systems at scale

• Ability to proactively learn new concepts and technology and apply them at work

• Skilled at solving ambiguous problems

• Strong collaboration and mentorship skills

• Masters in technical field such as computer science, mathematics, statistics or equivalent years of experience

• 10+ years of industry machine learning experience

• Experience working with machine learning, recommendation and ranking systems, or vector similarity search

Why Axion Ray

• Join a fast-growing, early-stage startup with the chance to make a significant impact from day one

• Be part of a world class engineering organization

• Our culture puts people first - we will encourage you to bring your full self to work and will set you up to succeed

• Competitive compensation, meaningful equity, and benefits package 

• Career development and leadership opportunities

• We’re an interdisciplinary team that prizes collaboration & diversity of thinking

• A chance to make a significant impact in improving the product quality and integrity in the manufacturing space

• Access to cutting-edge technology and resources

Axion is based in New York, New York with a headquarters in Williamsburg and the team works in a hybrid capacity. This role is expected to work in the office at least 3 days weekly.

Axion Ray is an Equal Opportunity / Affirmative Action employer committed to diversity in the workplace. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, age, national origin, disability, protected veteran status, gender identity, or any other factor protected by applicable federal, state, or local laws.