Senior Staff Machine Learning Engineer - Infrastructure, Automated Officiating
New York, NY, United States of America(US), 10022
WORK OPTION: The NBA currently provides eligible employees the option of working remotely one day per week.
Group Summary:
The Basketball Strategy & Growth department is responsible for data collection, analysis and technology pertaining to all on-court activities. The group, in partnership with Referee Operations, oversees the Game Review Program to help drive improvements in referee performance and rules clarification initiatives. Basketball Strategy & Growth also leads pivotal initiatives focused on innovating and improving the NBA game, such as rules changes, improvements to the competition format and implementation of technologies to improve player health, game integrity and fan engagement.
The Automated Officiating team is a new function within the Basketball Strategy & Growth department. This team is focused on innovating the on-court product through internally developed and deployed technologies. They spearhead key officiating technology initiatives from concept to launch, leveraging their cross-discipline expertise in real-time perception and sensing, computer vision, machine learning, and data analytics. The primary near-term focus of this team is deploying a system that can automatically detect and determine objective calls (e.g., out-of-bounds) in real-time during live NBA games.
Position Summary:
The NBA is seeking an experienced machine learning engineer to build machine learning infrastructure from the ground up and be a key contributor to the Automated Officiating team. This team sits within Basketball Strategy & Growth, and its primary goal is to develop advanced multi-modal automated officiating – leveraging computer vision and other sensing technologies – to enhance call accuracy, streamline game flow, and provide decision-making consistency and transparency. This is a small team that works like a startup within the NBA and provides significant opportunities for ownership and accelerated learning and growth.
Ideal candidates will bring considerable expertise in building large-scale computer vision training infrastructure and owning ML pipelines in production. This role will report to the Engineering Lead and play a critical role in taking our product from 0 to 1, leveraging expertise typically found in autonomous vehicles, robotics, AR/VR, or other real-time ML-driven systems.
Major Responsibilities
- Design, deploy, and maintain large, distributed ML training and inference clusters.
- Build and maintain scalable data pipelines that handle petabyte scale multi-modal sensor data, and own model training throughout the ML lifecycle.
- Profile, debug and implement tooling as needed for comprehensive analysis and optimize system performance.
- Have a strong sense of ownership and be excited to make technical contributions across the automated officiating system, e.g. sensor pipelines, ML data pipelines, training, modeling and evaluation pipelines etc.
- Provide technical guidance and mentorship to other engineers on the team.
- Be a guardian of the codebase and push for clean, well-tested and highly extensible code.
- Stay up to date on research and actively contribute new ideas.
- Research and test different training approaches including parallelization techniques and numerical precision trade-offs across different model scales.
Qualifications:
- Bachelor’s degree in Computer Science, Electrical Engineering, Math or related field (or equivalent experience).
- Strong grasp of latest techniques to optimize training and inference workloads.
- Demonstrated proficiency with distributed training frameworks to train large models.
- Knowledge of various cloud platforms (GCP, AWS, Azure) and their Machine Learning service offerings.
- Familiarity with containerization and orchestration frameworks (e.g., Kubernetes, Docker).
- Background working on distributed task management systems and scalable model serving & deployment architectures.
- Understanding of monitoring, logging, observability, and version control best practices for ML systems.
- Demonstrated proficiency building and deploying machine learning solutions to production.
- Exposure to the entire ML stack, from data pipelines to model inference, ideally for computer vision.
- Excellent problem-solving skills and adaptability in a fast-paced environment.
- Excellent communication and interpersonal skills.
Salary Range: $210,000 - $300,000
Job Posting Title: Senior Director
We Consider Applicants For All Positions On The Basis Of Merit, Qualifications And Business Needs, And Without Regard To Race, Color, National Origin, Religion, Sex, Gender Identity, Age, Disability, Alienage Or Citizenship Status, Ancestry, Marital Status, Creed, Genetic Predisposition Or Carrier Status, Sexual Orientation, Veteran Status, Familial Status, Status As A Victim Of Domestic Violence Or Any Other Status Or Characteristic Protected By Applicable Federal, State, Or Local Law.
The NBA is committed to providing a safe and healthy workplace. To safeguard our employees and their families, our visitors, and the broader community from COVID-19, and in consideration of recommendations from health authorities and the NBA’s own advisors, any individual working onsite in our New York and New Jersey offices must be fully vaccinated against COVID-19. The NBA will discuss accommodations for individuals who cannot be vaccinated due to a medical reason or sincerely held religious belief, practice, or observance.
Job Segment:
Computer Science, Electrical Engineering, Engineer, Learning, Electrical, Technology, Engineering, Human Resources