CrawlJobs Logo

Staff Machine Learning Engineer, Compute

United States, Austin 198900.00 - 304800.00 USD / Year · Job Posted March 03, 2026
Apply Position
Job Link Share

Job Description

Staff ML Engineer, ML Compute Platform. The ML Compute Platform is part of the AI Compute Platform organization within Infrastructure Platforms. Our team owns the cloud-agnostic, reliable, and cost-efficient compute backend that powers GM AI. We’re proud to serve as the AI infrastructure platform for teams developing autonomous vehicles (L3/L4/L5), as well as other groups building AI-driven products for GM and its customers. We enable rapid innovation and feature development by optimizing for high-priority, ML-centric use cases. Our platform supports the training and deployment of state-of-the-art (SOTA) machine learning models with a focus on performance, availability, concurrency, and scalability. We’re committed to maximizing GPU utilization across platforms while maintaining reliability and cost efficiency.

Job Responsibility

  • Design core platform backend software components
  • Thrive in a dynamic, multi-tasking environment with ever-evolving priorities
  • Interface with other teams to incorporate their innovations and vice versa
  • Analyze and improve efficiency, scalability, and stability of various system resources
  • Proactively identify, drive and design large initiatives across GM ML ecosystem

Requirements

  • 7+ years of industry experience
  • Expertise in either Go, C++, Python or other relevant coding languages
  • Strong background with kubernetes at scale
  • Relevant experience building large-scale with distributed systems
  • Experience leading and driving large scale initiatives
  • Experience working with Google Cloud Platform, Microsoft Azure, or Amazon Web Services

Nice to have

  • Hands-on experience in ML platforms
  • Experience with GPU/TPU optimizations
  • Experience with training frameworks like PyTorch, TorchX
  • Experience with Ray framework
  • Leadership/active participation in the open source community
  • Experience infrastructure applications or similar experience

What we offer

  • medical
  • dental
  • vision
  • Health Savings Account
  • Flexible Spending Accounts
  • retirement savings plan
  • sickness and accident benefits
  • life insurance
  • paid vacation & holidays
  • tuition assistance programs
  • employee assistance program
  • GM vehicle discounts
  • company vehicle evaluation program
  • incentive pay program

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Similar Jobs for

Staff Machine Learning Engineer, Compute

8 matching positions

New

Staff Machine Learning Engineer, Underwriting and Credit

Block has provided over $200 billion in credit to customers globally. Afterpay a...
Location
Location
United States , Bay Area
Salary
Salary:
276800.00 - 415200.00 USD / Year
block.xyz Logo
Block
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • A Bachelor's degree in a quantitative field (e.g., Mathematics, Statistics, Physics, Computer Science). Advanced degrees welcome.
  • 10+ years applying AI, machine learning, or statistical modeling in decisioning contexts such as credit, risk, fraud, recommendations, or similar domains.
  • Experience with probabilistic models and decision systems, including calibration, score transformations, and interpretation of model outputs.
  • Strong experimentation skills: you know how to design holdouts, measure lift, and evaluate models beyond aggregate metrics.
  • Experience with model monitoring, degradation detection, and retraining strategies in production systems.
  • Proficiency with AI-native development workflows. You use LLMs, agentic coding tools, and AI-assisted automation as a regular part of how you build and ship.
  • Experience explaining modeling concepts, results, and limitations to senior stakeholders and cross-functional partners.
  • Experience working across disciplines in environments with meaningful constraints.
Job Responsibility
Job Responsibility
  • Build, evaluate, and maintain underwriting and decisioning models across Cash App Borrow and Afterpay.
  • Design and evolve credit decision frameworks, including the modeling, automation, and policy logic that manage credit exposure over time.
  • Design and run experiments to evaluate model performance, measure impact on approval rates and loss, and inform credit policy decisions.
  • Develop deep understanding of borrower behavior, repayment dynamics, and portfolio structure across both products, and use that to inform model design and decision logic.
  • Contribute analysis and perspective that inform portfolio-level decisions, including explaining model behavior, tradeoffs, and uncertainty to senior technical and business leaders.
  • Work across the full modeling lifecycle: problem formulation, feature engineering, training, calibration, deployment, monitoring, and iteration in production.
  • Build agentic engineering workflows that accelerate development, testing, and documentation.
  • Collaborate with Product, Engineering, Legal, Compliance, and Operations to ensure credit systems reflect business goals and regulatory expectations.
  • Share modeling context and approaches across teams, helping align how credit risk is measured, interpreted, and discussed.
  • Shape how AI developer tooling is adopted across the team, defining review practices, quality standards, and governance patterns.
What we offer
What we offer
  • Remote work
  • medical insurance
  • flexible time off
  • retirement savings plans
  • modern family planning
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer - ML Training Infrastructure

The Role:   We are seeking an experienced, technically strong, impact-driven ex...
Location
Location
United States , Austin; Mountain View
Salary
Salary:
185000.00 - 335300.00 USD / Year
gm.com Logo
General Motors
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree or higher in Computer Science or a related field, or equivalent practical experience
  • 8+ years of professional software engineering experience
  • 5+ years of specialized experience in AI/ML infrastructure, such as enabling distributed training for large-scale ML models
  • Strong programming skills in Python, with deep proficiency in frameworks such as PyTorch (preferred), TensorFlow, or similar ML systems
  • Proven experience designing and operating distributed systems for ML training, including distributed computing, GPU computing, and cloud environments (AWS, GCP, Azure)
  • Demonstrated track record of leading technically ambiguous, cross-team infrastructure initiatives and driving them to measurable impact
  • Strong architectural judgment and ability to make sound technical tradeoffs across performance, reliability, usability, and cost
  • Willingness to travel to Sunnyvale, CA as needed
  • Comfortable operating in highly ambiguous and dynamic environments
Job Responsibility
Job Responsibility
  • Define and drive the architecture, design, and development of scalable, reliable, and high-performance ML frameworks and platform capabilities to support model training at scale
  • Lead model training performance analysis and optimization efforts across distributed training workflows, improving scalability, efficiency, and cost across heterogeneous hardware environments
  • Raise the bar on system observability, debuggability, operational excellence, and developer experience across the ML training stack
  • Own large, ambiguous, cross-functional technical initiatives from strategy through execution, including technical roadmap definition, tradeoff analysis, and delivery
  • Influence platform direction by identifying long-term infrastructure investments, setting engineering standards, and driving adoption of best practices across teams
  • Collaborate across organizational boundaries to align requirements, resolve technical disagreements, and integrate new capabilities into the platform ecosystem
  • Mentor engineers through design reviews, technical guidance, and hands-on partnership, while elevating engineering quality across the team
What we offer
What we offer
  • medical
  • dental
  • vision
  • Health Savings Account
  • Flexible Spending Accounts
  • retirement savings plan
  • sickness and accident benefits
  • life insurance
  • paid vacation & holidays
  • tuition assistance programs
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer - Rider Intelligence

Rider Experience drives and enables the critical trip booking funnel within the ...
Location
Location
United States , Seattle
Salary
Salary:
232000.00 - 258000.00 USD / Year
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Ph.D., M.S. or Bachelor in Computer Science, Mathematics with focus on Machine Learning, or equivalent technical background with exceptional demonstrated impact
  • 8+ years experience leading the development and deployment of ML models in large-scale production environments at top-tier ML companies
Job Responsibility
Job Responsibility
  • Define and execute technical strategies, spanning from model and system architecture to business objectives and stakeholder alignment
  • Lead the design, development, and production of end-to-end ML solutions for large-scale distributed systems serving billions of trips
  • Lead and mentor a team of Machine Learning Engineers (MLEs), providing technical leadership, setting the vision, and guiding the team through the full development lifecycle—from ideation to model deployment and scaling
What we offer
What we offer
  • Eligible to participate in Uber's bonus program
  • May be offered an equity award & other types of comp
  • Eligible to participate in a 401(k) plan
  • Various benefits
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer

The Marketplace Signals team at Uber is responsible for building and optimizing ...
Location
Location
United States , San Francisco; Sunnyvale
Salary
Salary:
232000.00 - 258000.00 USD / Year
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Ph.D. or M.S. in Statistics, Economics, Mathematics, Computer Science, Machine Learning, Operations Research, or other quantitative fields
  • 6+ years of industry experience in machine learning, including building and deploying ML models at scale
  • Experience in modern deep learning architectures and probabilistic modeling
  • Proficiency in programming languages (Python, Java, Scala) and ML frameworks (TensorFlow, PyTorch, Scikit-Learn)
  • Solid understanding of MLOps practices, including design documentation, testing, and source code management with Git
  • Advanced skills in the development and deployment of large-scale ML models and optimization algorithms
  • Strong business and product sense: ability to shape vague questions into well-defined analyses and success metrics that drive business decisions.
Job Responsibility
Job Responsibility
  • Develop and optimize ML models to enhance key marketplace signals (e.g., ETA predictions, supply availability metrics, demand forecasts)
  • Collaborate with cross-functional teams (Pricing, Matching, Driver Incentives, etc.) to ensure marketplace signals are effectively utilized
  • Improve operational efficiency by building a centralized, scalable system for marketplace signals that serves multiple use cases
  • Leverage cutting-edge ML techniques (deep learning, probabilistic modeling, reinforcement learning, etc.) to continuously refine marketplace signals
  • Ensure consistency and reliability across Uber's platform by maintaining high-quality marketplace signals that inform rider and driver experiences
  • Reduce technical debt by streamlining signal infrastructure and minimizing redundant computations
  • Work with real-time streaming data and large-scale distributed systems to ensure Uber's signals are up-to-date and responsive to market dynamics.
What we offer
What we offer
  • Bonus program
  • Equity award
  • 401(k) plan
  • Various benefits
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer, Credit Products (Square Financial Services)

The Credit and Lending team is responsible for the predictive intelligence that ...
Location
Location
United States , Seattle
Salary
Salary:
194500.00 - 343100.00 USD / Year
block.xyz Logo
Block
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Minimum of 8 years of related experience with a Bachelor's degree
  • or 6 years and a Master's degree
  • or a PhD with 3 years experience, with a focus on developing and deploying machine learning and statistical models in production environments
  • A degree in a technical field (e.g., Computer Science, Mathematics, Statistics, Physics, or Engineering)
  • Strong quantitative intuition and data visualization skills
  • Full-stack proficiency preferred
  • The versatility to communicate clearly with both technical and non-technical audiences
  • A pragmatic approach to problem-solving
Job Responsibility
Job Responsibility
  • Apply a rigorous scientific mindset to the challenge of underwriting new customer segments
  • Lead complex ML Operations and Infrastructure initiatives
  • Design and implement the full credit modeling stack
  • Use data science techniques to leverage new data sources for modeling
  • Identify and execute material improvements to credit policy
  • Support team members in ad-hoc and scheduled updates to existing models
  • Operate effectively within the framework of a regulated bank
What we offer
What we offer
  • Remote work
  • medical insurance
  • flexible time off
  • retirement savings plans
  • modern family planning
  • Fulltime
Read More
Arrow Right

Sr Staff Machine Learning Engineer (ADEM)

The ADEM engineering team is the engine of innovation at Palo Alto Networks. We ...
Location
Location
United States , Santa Clara
Salary
Salary:
147000.00 - 237500.00 USD / Year
paloaltonetworks.com Logo
Palo Alto Networks
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 6+ years of software engineering experience, with a significant portion dedicated to designing and operating complex, resource-constrained client-side or endpoint software in a large-scale deployment
  • As a leader in engineering efficiency, you must be an expert in leveraging AI-powered development tools—including Claude Code, Cursor, Windsurf, and GitHub Copilot—to radically accelerate the SDLC and automate complex refactoring and testing workflows
  • Proven track record of architecting and shipping production-quality software for multiple operating systems (Windows, macOS, Linux) with a deep understanding of OS internals, networking stacks (L3-L7), and system performance tuning
  • Coding Rust, or Go (Golang) for high-performance client development is required
  • BS/MS/PhD in Computer Science or a related technical field, or equivalent high-level professional experience.
Job Responsibility
Job Responsibility
  • Define the long-term architectural vision for the ADEM Agent across Linux, Windows, macOS, and mobile platforms, ensuring robust performance, minimal resource consumption
  • Lead the 'Design Review Board' for all Agent-side features, ensuring all components adhere to Secure AI by Design, operate with high-availability patterns in diverse OS environments, and maintain a low CPU/memory footprint
  • Drive the evolution of the Agent's telemetry collection and processing pipeline, optimizing for low-latency, encrypted data transfer and ensuring resilience in low-bandwidth or disconnected environments
  • Partner with Product Management, Core Infrastructure, and Security teams to translate ambiguous business requirements into robust, high-performance technical specifications for the client software
  • Act as a force multiplier by mentoring Staff and Senior engineers, fostering a culture of rigorous platform-specific testing, high code quality, and proactive technical debt management within the Agent codebase
  • Serve as the ultimate technical escalation point for complex, systemic production issues related to endpoint stability or performance, leading post-mortems that drive permanent architectural improvements in the Agent
What we offer
What we offer
  • restricted stock units
  • bonus
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer - AV Labs

Uber is launching AV Labs to accelerate the autonomous technology ecosystem. We'...
Location
Location
United States , Sunnyvale
Salary
Salary:
232000.00 - 258000.00 USD / Year
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 8+ years of working experience in the ML, Robotics, or Autonomous Systems industry
  • Proven experience leading large-scale technical projects from conception to production
  • Bachelor’s degree in Computer Science, Computer Engineering, or related fields
  • Expert-level proficiency in Python and Linux environments
  • Deep expertise in modern AI/ML frameworks (e.g., PyTorch)
Job Responsibility
Job Responsibility
  • Set the technical roadmap and lead the delivery of state-of-the-art Machine Learning systems
  • Algorithm Development: Lead the strategy for development of autonomy algorithms and foundation models that extract high-fidelity semantic meaning from complex urban edge cases to enrich our L4 data lake
  • Architecture & System Design: Design and oversee the implementation of complex, large-scale ML systems, ensuring seamless integration between upstream sensor data
  • Technical Mentorship & Influence: Mentor senior and lead engineers, fostering a culture of rigorous experimentation and engineering excellence
  • Platform Evolution: Define the requirements for high-quality datasets and auto-labeling systems, ensuring our ML infrastructure evolves at the speed of the latest research
  • Cross-Organizational Leadership: Act as a bridge between AV Labs and other Uber engineering units to ensure that autonomous technology is successfully integrated and deployed at scale
What we offer
What we offer
  • Eligible to participate in Uber's bonus program
  • May be offered an equity award & other types of comp
  • Eligible to participate in a 401(k) plan
  • Various benefits
  • Fulltime
Read More
Arrow Right

Staff Machine Learning Engineer, Credit Products (Square Financial Services)

Block is one company built from many blocks, all united by the same purpose of e...
Location
Location
United States , Bay Area
Salary
Salary:
Not provided
block.xyz Logo
Block
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Minimum of 8 years of related experience with a Bachelor's degree
  • or 6 years and a Master's degree
  • or a PhD with 3 years experience, with a focus on developing and deploying machine learning and statistical models in production environments
  • A degree in a technical field (e.g., Computer Science, Mathematics, Statistics, Physics, or Engineering)
  • Strong quantitative intuition and data visualization skills, with a proven ability to conduct sophisticated ad-hoc and exploratory analysis
  • Full-stack proficiency preferred, including the ability to contribute across the entire technical stack—from data pipelines to production-grade software architecture
  • The versatility to communicate clearly with both technical and non-technical audiences, particularly in the context of high-visibility projects and executive stakeholders
  • A pragmatic approach to problem-solving, with a willingness to utilize whichever tool is most appropriate for the situation while balancing complex business, technical, and regulatory constraints
  • Experience with tree-based models and gradient boosting is helpful but not required
  • we value the ability to adapt and learn new methodologies as the credit landscape evolves
Job Responsibility
Job Responsibility
  • Apply a rigorous scientific mindset to the challenge of underwriting new customer segments, involving the evaluation of alternative external data sources and the deployment of advanced architectures to enhance predictive accuracy
  • Lead complex ML Operations and Infrastructure initiatives that advance our modeling capabilities, such as scaling data ingestion or enabling the use of more complex neural networks
  • Design and implement the full credit modeling stack, taking responsibility for the entire lifecycle of credit decisioning and ensuring models are robustly integrated into production environments
  • Use data science techniques to leverage new data sources for modeling, making sense of messy datasets and bringing clarity to business decisions
  • Identify and execute material improvements to credit policy, applying an analytical lens to determine where technical or logic shifts can yield significant positive outcomes for the customer and the bank's portfolio
  • Support team members in ad-hoc and scheduled updates to existing models, and help troubleshoot issues in a real-time production environment
  • Operate effectively within the framework of a regulated bank (SFS), balancing rapid innovation with the requirements of safety, soundness, and compliance
What we offer
What we offer
  • Remote work
  • Medical insurance
  • Flexible time off
  • Retirement savings plans
  • Modern family planning
  • Fulltime
Read More
Arrow Right