Full-Time

Machine Learning Infrastructure Engineer

Updated on 9/4/2026

xAI

xAI

1,001-5,000 employees

Generative AI chatbot leveraging real-time data

Compensation Overview

$180k - $440k/yr

+ Equity

Palo Alto, CA, USA

In Person

Bachelor's, Master's, PhD

Category
DevOps & Infrastructure (1)
Required Skills
Graphics Processing Unit (GPU)
Rust
Python
Puppet
Distributed Systems
Neural Networks
CUDA
PyTorch
Machine Learning
Computer Networking
Data Engineering
Ansible
C/C++
Linux/Unix

Get referred to xAI

See people who can refer or advise you

Requirements
  • A Bachelor, Master, post-graduate, or PhD degree in computer science, machine learning, or another quantitative discipline, or equivalent work experience.
  • At least 2 years of industry experience working with high-traffic or large-scale production environments, distributed systems, GPU infrastructure, and/or deep learning applications.
  • At least 2 years of experience with ML platforms, training infrastructure, or close collaboration with modeling engineers and data scientists.
  • Strong proficiency with Python and experience with compiled languages such as C++ or Rust.
  • Familiarity with modern machine learning frameworks such as JAX or PyTorch.
  • Low-level understanding of compute systems, including distributed storage, NVIDIA drivers, CUDA toolkits, and networking.
  • Familiarity with Linux systems and orchestration tools.
  • Experience with job schedulers such as Slurm, configuration management tools such as Puppet or Ansible, or related infrastructure tooling.
Responsibilities
  • Design, build, and scale GPU compute infrastructure, training frameworks, and experimentation tools to enable rapid iteration on machine learning hypotheses.
  • Develop data pipelines and integrate large-scale data, training, and inference systems.
  • Collaborate with machine learning teams to productionize models and ensure seamless integration across the stack.
  • Ensure the scalability, reliability, and efficiency of large-scale machine learning systems.
  • Work across the full stack to solve complex problems independently.
  • Mentor junior engineers and contribute to the growth of the team.
Desired Qualifications
  • Deep familiarity with modern machine learning frameworks such as JAX or PyTorch.
  • Low-level understanding of compute systems, including distributed storage, NVIDIA drivers, CUDA toolkits, and networking.
  • Comfort with Linux systems and orchestration tools.
  • Experience with job schedulers such as Slurm, configuration management tools such as Puppet or Ansible, or related infrastructure tooling.

xAI builds Grok, a generative AI chatbot with a Hitchhiker’s Guide-inspired persona that uses real-time data from X to produce current, culturally aware responses. Grok is accessible to premium subscribers on X, via a standalone website, mobile apps, and through an API for developers. The company also develops large-scale AI infrastructure, such as the Colossus supercomputer, to train and run its models. Its goal is to pursue truth-seeking AGI research and grow a capital-intensive platform that could support a major public offering in the future.

Company Size

1,001-5,000

Company Stage

Series E

Total Funding

$42.1B

Headquarters

Palo Alto, California

Founded

2023

Get referred to xAI

See people who can refer or advise you

Simplify Jobs

Simplify's Take

What believers are saying

  • xAI raised $20 billion on January 6, 2026 from Nvidia, Cisco, and sovereign backers.
  • Grok 4.3 and Grok 4.5 launched in 2026, showing fast product cadence.
  • February 2026 SpaceX acquisition gave xAI deeper distribution, infrastructure, and fundraising firepower.

What critics are saying

  • NAACP sued xAI on April 14, 2026 over 27 unpermitted Southaven turbines.
  • Reuters reported June 9, 2026 residents filed a class action over turbine noise and health harms.
  • If courts shut Southaven turbines, Grok training stalls and xAI loses its compute advantage.

What makes xAI unique

  • XAI-owned Grok 4.3 uses real-time X, web, and news search for live answers.
  • Musk merged xAI into SpaceX in February 2026, pooling rockets, satellites, and capital.
  • Colossus in Memphis gives xAI unusually large training compute for rapid model iteration.

Help us improve and share your feedback! Did you find this helpful?

Benefits

Health Insurance

Remote Work Options

Growth & Insights and Company News

Headcount

6 month growth

11%

1 year growth

2%

2 year growth

11%
Bitcoin.com
Sep 3rd, 2026
A 52-year-old Mississippi farmer faces elon musk's data-center push.

A 52-year-old Mississippi farmer faces elon musk's data-center push. A SpaceXAI-backed buildout of massive data centers in Southaven, Mississippi, near the Tennessee border is rattling nearby residents with round-the-clock noise from gas turbines and fears of toxic emissions. The NAACP is pursuing legal action against SpaceXAI subsidiary MZX Tech, while a community petition has gathered more than 1,400 signatures as families report worsening asthma. Key takeaways. * SpaceXAI's 69 gas turbines sparked noise and pollution fears near Southaven homes. * NAACP legal action and 1,400 petition signatures are mounting pressure on MZX Tech. * SpaceXAI plans to replace 69 temporary turbines with 41 permanent structures. On the edge of Southaven, Mississippi, where family acreage still carries four generations of decisions, a new neighbor has moved in: SpaceXAI's data-center buildout near the Tennessee line. For 52-year-old farmer Krystal Polk, the tradeoffs are no longer abstract, with gas turbines running around the clock and a soundscape she likens to being boxed in by planes heading for the airport. Residents say the air is changing too, and the Polk family's asthma has become part of the project's real-world cost. Now the fight is spreading beyond property lines, with the NAACP suing SpaceXAI's subsidiary MZX Tech and a petition that has topped 1,400 signatures. A state-of-the-art tech hub, a struggling community. Along the Mississippi-Tennessee line, a quiet stretch of Southaven has become a front-row seat to America's AI infrastructure buildout. A project tied to SpaceXAI, described locally as a major data center installation, is bringing industrial-scale power generation and round-the-clock operations into a residential farming area. The promise is more compute. The daily reality, neighbors say, is constant noise and worsening air. What's striking is how fast "AI capacity" turns physical: fuel deliveries, construction crews, and turbines that never seem to rest. Residents describe the sound as a persistent low roar, the kind that seeps into sleep and conversation. For a community that prized stillness and space, the new baseline feels like living beside infrastructure built for somewhere else. Krystal Polk's land, and the cost of proximity. At the center is Krystal Polk, a 52-year-old farmer whose family has held roughly 2.5 acres across 4 generations. She told CBS News the noise resembles an airport that never quite takes off. The land is more than a parcel, she argues, it is a hard-won foothold in a region where property ownership has long been uneven. Polk says the family's health has deteriorated as the site has expanded, including more frequent asthma-related episodes. Neighbors echo similar complaints, linking the changes to heavier air pollution and continuous operation. When your home is your livelihood, how do you measure the loss of a place that no longer feels livable? Legal pressure and a community that won't stay quiet. Organizing has accelerated. The NAACP and other groups have pursued legal action against MZX Tech, described as a subsidiary tied to the project, alleging negligence around air and noise impacts. A petition calling for stronger environmental practices has drawn 1,400 plus signatures, according to local organizers. The company has pointed to mitigation plans, including investments in noise-reducing barriers. Residents say those steps have not matched the scale of the problem, especially with operations continuing day and night. The fight is becoming a familiar American story: communities learning, in real time, how to challenge the machinery behind the cloud. From "temporary" turbines to permanent change. One detail keeps coming up in local conversations: the energy setup. Reports describe 69 temporary gas turbines that are expected to be replaced by 41 permanent structures, a shift that may reduce some strain but also signals long-term commitment. Independent analyses cited by advocates warn that air-quality risks can remain significant near residential areas. Meanwhile, some farmers say they are facing pressure to sell, with real estate interests circling land near the development. SpaceXAI has also leaned on connections to national-security AI work as part of its posture, complicating the path for local challenges. For Polk, it is not a symbolic debate about technology. It is a bid to keep a home, and a family history, from being priced out by the servers next door. Yesterday Greed Last Week Greed Last Month Fear How do you feel about the market today?

GlobeNewswire
Sep 2nd, 2026
Coder and SpaceXAI collaborate to bring agentic coding inside the world's most regulated enterprises.

Coder and SpaceXAI collaborate to bring agentic coding inside the world's most regulated enterprises. Coder Agent Relay lets banks, government agencies, and global enterprises run Cursor's cloud agents entirely on their own infrastructure, opening a market that vendor-hosted tools have never been able to serve. AUSTIN, Texas, Sept. 02, 2026 (GLOBE NEWSWIRE) - Coder today announced Coder Agent Relay, a self-hosted execution environment for cloud coding agents, with SpaceXAI as its launch partner. Cursor Cloud Agents can now run inside Coder workspaces on infrastructure the customer already operates. Developers keep the Cursor experience they know - app, web, and mobile - while Cursor continues to run the agent loop, including inference and planning. Tool calls execute in Coder environments on the customer's network, so source code, secrets, and internal services stay on machines they control. The market AI coding has not been able to reach Demand for AI coding agents inside large enterprises has exploded, and adoption in regulated industries hasn't kept pace, but not for the reason most people assume. Developers in these organizations want the same tools everyone else has. The blocker is the deployment model. Banks, defense agencies, government institutions, and global enterprises operate under requirements that vendor-hosted tools can't satisfy: source code needs controlled access, execution environments need security and governance that prevent data exfiltration, and every action needs to be auditable. Those requirements have kept the best AI coding tools out of the environments that stand to gain the most from them. Gartner projects that 80% of enterprise software engineers will need to upskill for generative AI by 2027. In the most regulated sectors, security and compliance review still decide which tools ever reach a developer. That is precisely the market Coder opens. Coder is the self-hosted infrastructure the world's most demanding organizations already trust to run development inside their own walls. Pairing that foundation with SpaceXAI's agentic experience turns a market that has been closed to vendor-hosted AI tools into one the two companies can finally enter together. AI needs an operating layer The modern enterprise stack runs in layers: application, data, compute. AI fits none of them. It touches all three and belongs to none, which is why organizations everywhere hit the same wall the moment AI moves from pilots into production. The AI Operating Layer is the layer AI requires: where AI runs inside the organization, on its own infrastructure, under its own policies, visible to its own teams. Built right, it enforces five things: data stays inside the boundary, every task sees only what it was granted, only approved models run, every action leaves a record, and spend is capped before it accumulates. That is what ends the trade-off AI programs keep being forced into, speed or safety. Teams adopt whatever tools make them fastest, agents work with real autonomy, and the layer underneath holds. The collaboration extends Coder's ability to deliver: Cursor's agents now execute within the enterprise-controlled governed environments, bringing autonomous development to the industries where the layer matters most. "Enterprises never rejected AI agents. They rejected the deployment model. The model was never the bottleneck. What has kept this work out of production is everything that comes after the pilot: where the code runs, who has access, what the agent can reach, and what record exists afterward. Those are infrastructure questions, and Coder was built to answer them. Cursor builds a coding experience developers love. We give it the ground to run on inside a bank, an agency, or a defense program," said Rob Whiteley, CEO of Coder. What Coder Agent Relay delivers The collaboration pairs a joint go-to-market motion with a direct product integration that gives enterprises greater security, governance, and control over cloud-hosted coding agents. Each Coder workspace can start a Cursor worker that opens an outbound connection to Cursor. Platform teams can provision and scale those workspaces the same way they already manage developer environments, so regulated organizations can deploy Cloud Agents at scale next to private code and custom hardware without giving up Cursor's product surface or the security and standardization of Coder workspaces. * Your perimeter, your rules. Workspaces run on the customer's cloud, VPC, or on-premises environment. Code does not leave, and network egress policy is set once by the platform team and enforced uniformly across every workspace, developer or agent. Air-gapped deployment is available where no external access is acceptable. * Boundaries by architecture, not by trust. Agent environments are sandboxed, ephemeral, and scoped to a single task. A prompt injection that would push an agent toward unauthorized resources is blocked at the environment layer, not left to the model to refuse. * Policy as infrastructure. Approved models, permitted data sources, and available resources are defined once at the environment level and inherited by every workspace that follows, without per-team reconfiguration. * A complete record. Every run produces a log of what the agent accessed, executed, changed, and was blocked from doing, so compliance reporting for any time window does not have to be reconstructed by hand. "Engineering teams want to build with Cursor. Enterprise security policies require that sensitive information not leave its own infrastructure. Coder provides those teams a path to run Cursor inside environments they fully control, giving developers the experience they want while maintaining the security controls they need," said Toni Adams, Sr Director, Partnerships at SpaceXAI. The foundation for enterprise AI development SpaceXAI is the first of several partnerships Coder is building around a simple idea: the industry's best AI coding tools should run inside any organization, on any infrastructure, without the organization giving up control to get them. SpaceXAI is the first agent provider with Agent Relay, and the integration is designed to extend self-hosted execution to more over time. Enterprises have been here before. When cloud moved infrastructure out of the building, companies stood up platform teams to govern how it was provisioned, secured, and paid for. AI is forcing the same mandate now, with a twist: much of the work is coming back inside the organization's own walls. The principle hasn't changed; only the technology has. Coder is where enterprises build the layer it runs on. Availability Coder Agent Relay for Cursor is currently in private preview with design partners. If security requirements have kept Cursor out of reach, visit coder.com/cursor for docs and more information about bringing it inside your enterprise. About Coder Coder is the AI Operating Layer for the enterprise: self-hosted, model- and cloud-agnostic infrastructure that runs AI inside the organization's own environment, under its own policies, and in full view of its own teams. Global 2000 enterprises and government agencies use Coder to give developers and autonomous agents one governed workspace, moving source code off laptops and onto centrally managed infrastructure that platform teams can secure, audit, and scale. With more than 125,000 GitHub stars and trusted by the world's most regulated organizations, Coder is built for the agentic era. Learn more at coder.com.

Augeo Marketing
Sep 2nd, 2026
The frontier of loyalty: Augeo leverages SpaceXAI to launch 100% personalization across every audience an enterprise serves.

The frontier of loyalty: Augeo leverages SpaceXAI to launch 100% personalization across every audience an enterprise serves. September 2, 2026 THEO (Trusted Human Engagement Orchestrator), Augeo's agentic AI orchestration core, is now integrating Grok's real-time reasoning and X's live sentiment data - powering the first 100% personalized redemption and offers experience and the first-of-its-kind, Loyalty Media Network. ST. PAUL, MN (September 2, 2026) - Augeo, a global leader in loyalty and engagement technologies, today announced its plan to reimagine how enterprises connect with the people they serve. Collaborating with SpaceXAI to integrate Grok's real-time reasoning and X's live sentiment data into THEO, Augeo's orchestration core, Augeo is building two industry firsts: the first 100% personalized redemption and offers experience and the first-of-its-kind Loyalty Media Network spanning loyalty rewards, offers, and social advocacy. Augeo's new offerings will advance a bold vision for enterprise-wide engagement - a single, unified approach to motivating every audience an organization serves. From employee engagement to customer loyalty to advertising through the new Loyalty Media Network, Augeo reaches every audience: employees, customers, dealers, channel partners, fans and members. Rather than running siloed programs, enterprises can engage their entire ecosystem through one intelligent core. Wherever loyalty lives, Augeo is there. That reach is the point. Workplace, marketplace, and media were once separate disciplines served by separate vendors. Augeo brings them together under one roof and one intelligence core - so when loyalty and engagement strategies are on the agenda, in any form, Augeo is the company to call. Let's talk! THEO: Augeo's orchestration core. THEO is the intelligence layer that unifies data, decisioning and delivery into a single system. With Grok's real-time reasoning and X's live sentiment data, THEO understands each individual in the moment and acts inside the channels and systems enterprises already use, no migration, no rip-and-replace. The result is engagement that meets people where they already are and gets sharper with every interaction. Critically, it does so with a commitment to privacy and zero data retention. Two industry firsts. The first 100% personalized redemption and offers experience. Every offer, reward and interaction is tailored precisely to the individual, built for the moment, rather than aimed at a segment. The collaboration between Augeo's loyalty expertise and SpaceXAI makes personalization at this depth and scale possible for the first time. The first Loyalty Media Network. Retail media networks reach shoppers. The Augeo Loyalty Media Network goes further, giving brands a direct line to the most loyal, highest-value customers at the moment they are most receptive. It is a new category of media, born from loyalty and powered by real-time intelligence. "This is a ground-up collaboration to reimagine our entire sector. We're delivering two things the industry has never seen: a redemption and offers experience that is 100% personal to each individual, which only AI enables, and an imaginative revenue producing media network where there was not one before, all while processing real-time sentiment at global scale with a firm commitment to privacy and zero data retention. Enterprises don't have to choose all or nothing. They can begin in one area of the enterprise and extend from there, compounding retention, revenue and reach as they grow. This is how brands and people will connect going forward." -David Kristal, Founder and CEO of Augeo THEO is available beginning this fall to Augeo's enterprise client base. To learn more, connect with Augeomarketing to have a conversation. About Augeo. Augeo is a global leader in loyalty and engagement technologies, helping define the new frontier of loyalty, the measurable connection between a person and a brand, employer, team or community. For 27 years, Augeo has provided the specialized infrastructure to drive performance across brand commerce, workforce recognition and fan engagement, serving hundreds of clients, including dozens of Fortune 500 companies, and fostering deeper connections for millions of people worldwide. Media Contact Alexandra Kolodge Associate Marketing Director 605.351.8953

Digital Terminal
Aug 31st, 2026
SpaceXAI appoints Manish Jain as Regional VP - India GTM.

SpaceXAI appoints Manish Jain as Regional VP - India GTM. Based in Delhi, he joins SpaceXAI after a recent stint at Wiz, where he was part of the company's founding India team, and more than 11 years at Palo Alto Networks. Published on: 31 Aug 2026, 10:48 pm SpaceXAI has appointed Manish Jain as a founding team member and Regional Vice President - India GTM, as the company begins building its India business and expanding its engagement with enterprises, customers and technology partners. Jain brings more than a decade of experience in enterprise technology and cybersecurity sales, channel development and go-to-market strategy. Based in Delhi, he joins SpaceXAI after a recent stint at Wiz, where he was part of the company's founding India team, and more than 11 years at Palo Alto Networks. His appointment comes as India's enterprise technology market enters a new phase of AI adoption, with organisations increasingly looking beyond experimentation toward deploying AI at scale. For AI companies, this is creating demand not only for advanced technology but also for strong local leadership capable of building customer relationships, partner ecosystems and sustainable go-to-market models. Building SpaceXAI's India Presence In his new role, Jain will focus on developing SpaceXAI's India go-to-market strategy and establishing the company's presence among enterprises and technology partners. He will work with Midhun Chandra Babu on the India GTM initiative and collaborate with the wider APJ team as SpaceXAI expands its regional footprint. The mandate will involve building relationships across the enterprise ecosystem while identifying opportunities to position SpaceXAI's AI capabilities for organisations looking to accelerate software development, innovation and business transformation. Jain's experience in building businesses from an early stage is expected to be particularly relevant as SpaceXAI develops its India operations. Sharing his appointment on LinkedIn, Manish Jain said, "I've joined SpaceXAI as a founding team member and Regional Vice President for India GTM. India won't just adopt AI this decade. It will export it. That isn't hype. It's a structural reality. India writes a massive share of the world's software, operates under relentless pressure to move faster, and holds an incredible talent pool waiting for true leverage." From Wiz to SpaceXAI Before joining SpaceXAI, Jain served as Regional Sales Director - India North and West at Wiz from April 2025 to August 2026. During his tenure, he was part of the founding India team and played a role in building the company's business from the ground up. A Long Enterprise Technology Journey Jain previously spent more than 11 years at Palo Alto Networks, where he held multiple sales leadership positions. His responsibilities covered enterprise, digital-native, mid-market and SMB customers, along with BFSI, public sector and service provider accounts. His regional experience extended across North and East India as well as Nepal, Bhutan and Bangladesh, giving him exposure to diverse customer segments and technology markets. This combination of direct sales leadership, regional GTM execution and partner development provides SpaceXAI with an experienced commercial leader as it establishes its India operations. India's Opportunity in AI With Jain joining as a founding team member, SpaceXAI is bringing in a leader with experience in building enterprise businesses, developing partner ecosystems and establishing technology brands in competitive markets. His mandate will be central to shaping the company's India GTM strategy as it looks to turn the country's growing AI momentum into a broader customer and partner ecosystem. [Stay informed with our latest updates by joining the] [WhatsApp Channel now!]

Fortune
Aug 31st, 2026
Elon Musk's X integrates MoonPay crypto payments into Grok AI chatbot globally

MoonPay has integrated its AI-enabled payment tool PayBox into Grok, X's chatbot, enabling users globally to conduct cryptocurrency transactions across multiple blockchain networks. Users can buy, lend, and move crypto through the service, which also supports non-crypto activities like booking flights. The integration includes Kamino, a Solana-based lending platform, allowing users to earn yield on USDC stablecoin. Users approve transactions with a passkey after Grok prepares them. This launch follows MoonPay's recent addition of crypto lending features for ChatGPT and Claude users outside the US, UK, EU, and Australia. The move reflects a broader industry trend toward "agentic commerce," with companies like Coinbase, Stripe, Visa, and Mastercard developing similar AI-powered payment tools.