ML Infrastructure Engineer

Back to jobs

New

ML Infrastructure Engineer

Palo Alto, California, United States

SpaceXAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity. We operate with a flat organizational structure. All employees are expected to be hands-on and to contribute directly to the company’s mission. Leadership is given to those who show initiative and consistently deliver excellence. Work ethic and strong prioritization skills are important. All employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

About The Role

As an ML Infrastructure Engineer, you will play a pivotal role in building and optimizing the reliable, high-performance ML platform that powers recommendations on X. We're looking for exceptional engineers who are passionate about our mission and have a strong desire to make a meaningful impact.

Responsibilities

• Designing, building, and scaling GPU compute infrastructure, training frameworks, and experimentation tools to enable rapid iteration on ML hypotheses
• Developing data pipelines and integrating large-scale data, training, and inference systems
• Collaborating with ML teams to productionize models and ensure seamless integration across the stack
• Ensuring scalability, reliability, and efficiency of large-scale machine learning systems
• Working across the full stack to solve complex problems independently
• Mentoring junior engineers and contributing to the growth of the team
  
  

Basic Qualifications

• Bachelor, Master, Post-graduate or PhD in computer science, machine learning, or other quantitative discipline; or equivalent work experience
• 2 years of industry experience working with high traffic or large-scale production environments, distributed systems, GPU infrastructure, and/or deep learning applications
• 2 years 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
  
  

Preferred Skills And Experience

• Deep familiarity with modern ML frameworks such as JAX or PyTorch
• Low-level understanding of compute systems, including distributed storage, NVIDIA drivers, CUDA toolkits, and networking
• Comfortable with Linux systems and orchestration tools
• Experience with job schedulers (e.g., Slurm), configuration management (Puppet/Ansible), or related infrastructure tooling
  
  

Compensation And Benefits

$180,000 - $440,000 USD

Base salary is just one part of our total rewards package at xAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.

SpaceXAI is an equal opportunity employer. For details on data processing, view our Recruitment Privacy Notice.

Create a Job Alert

Interested in building your career at SpaceXAI? Get future opportunities sent straight to your email.

Create alert

Apply for this job

indicates a required field

First Name*

Last Name*

Email*

Phone

Country*

Phone*

244 results found

No results found

• Afghanistan 93
• Åland Islands 358
• Albania 355
• Algeria 213
• American Samoa 1
• Andorra 376
• Angola 244
• Anguilla 1
• Antigua & Barbuda 1
• Argentina 54
• Armenia 374
• Aruba 297
• Ascension Island 247
• Australia 61
• Austria 43
• Azerbaijan 994
• Bahamas 1
• Bahrain 973
• Bangladesh 880
• Barbados 1
• Belarus 375
• Belgium 32
• Belize 501
• Benin 229
• Bermuda 1
• Bhutan 975
• Bolivia 591
• Bosnia & Herzegovina 387
• Botswana 267
• Brazil 55
• British Indian Ocean Territory 246
• British Virgin Islands 1
• Brunei 673
• Bulgaria 359
• Burkina Faso 226
• Burundi 257
• Cambodia 855
• Cameroon 237
• Canada 1
• Cape Verde 238
• Caribbean Netherlands 599
• Cayman Islands 1
• Central African Republic 236
• Chad 235
• Chile 56
• China 86
• Christmas Island 61
• Cocos (Keeling) Islands 61
• Colombia 57
• Comoros 269
• Congo - Brazzaville 242
• Congo - Kinshasa 243
• Cook Islands 682
• Costa Rica 506
• Côte d’Ivoire 225
• Croatia 385
• Cuba 53
• Curaçao 599
• Cyprus 357
• Czechia 420
• Denmark 45
• Djibouti 253
• Dominica 1
• Dominican Republic 1
• Ecuador 593
• Egypt 20
• El Salvador 503
• Equatorial Guinea 240
• Eritrea 291
• Estonia 372
• Eswatini 268
• Ethiopia 251
• Falkland Islands 500
• Faroe Islands 298
• Fiji 679
• Finland 358
• France 33
• French Guiana 594
• French Polynesia 689
• Gabon 241
• Gambia 220
• Georgia 995
• Germany 49
• Ghana 233
• Gibraltar 350
• Greece 30
• Greenland 299
• Grenada 1
• Guadeloupe 590
• Guam 1
• Guatemala 502
• Guernsey 44
• Guinea 224
• Guinea-Bissau 245
• Guyana 592
• Haiti 509
• Honduras 504
• Hong Kong SAR China 852
• Hungary 36
• Iceland 354
• India 91
• Indonesia 62
• Iran 98
• Iraq 964
• Ireland 353
• Isle of Man 44
• Israel 972
• Italy 39
• Jamaica 1
• Japan 81
• Jersey 44
• Jordan 962
• Kazakhstan 7
• Kenya 254
• Kiribati 686
• Kosovo 383
• Kuwait 965
• Kyrgyzstan 996
• Laos 856
• Latvia 371
• Lebanon 961
• Lesotho 266
• Liberia 231
• Libya 218
• Liechtenstein 423
• Lithuania 370
• Luxembourg 352
• Macao SAR China 853
• Madagascar 261
• Malawi 265
• Malaysia 60
• Maldives 960
• Mali 223
• Malta 356
• Marshall Islands 692
• Martinique 596
• Mauritania 222
• Mauritius 230
• Mayotte 262
• Mexico 52
• Micronesia 691
• Moldova 373
• Monaco 377
• Mongolia 976
• Montenegro 382
• Montserrat 1
• Morocco 212
• Mozambique 258
• Myanmar (Burma) 95
• Namibia 264
• Nauru 674
• Nepal 977
• Netherlands 31
• New Caledonia 687
• New Zealand 64
• Nicaragua 505
• Niger 227
• Nigeria 234
• Niue 683
• Norfolk Island 672
• North Korea 850
• North Macedonia 389
• Northern Mariana Islands 1
• Norway 47
• Oman 968
• Pakistan 92
• Palau 680
• Palestinian Territories 970
• Panama 507
• Papua New Guinea 675
• Paraguay 595
• Peru 51
• Philippines 63
• Poland 48
• Portugal 351
• Puerto Rico 1
• Qatar 974
• Réunion 262
• Romania 40
• Russia 7
• Rwanda 250
• Samoa 685
• San Marino 378
• São Tomé & Príncipe 239
• Saudi Arabia 966
• Senegal 221
• Serbia 381
• Seychelles 248
• Sierra Leone 232
• Singapore 65
• Sint Maarten 1
• Slovakia 421
• Slovenia 386
• Solomon Islands 677
• Somalia 252
• South Africa 27
• South Korea 82
• South Sudan 211
• Spain 34
• Sri Lanka 94
• St. Barthélemy 590
• St. Helena 290
• St. Kitts & Nevis 1
• St. Lucia 1
• St. Martin 590
• St. Pierre & Miquelon 508
• St. Vincent & Grenadines 1
• Sudan 249
• Suriname 597
• Svalbard & Jan Mayen 47
• Sweden 46
• Switzerland 41
• Syria 963
• Taiwan 886
• Tajikistan 992
• Tanzania 255
• Thailand 66
• Timor-Leste 670
• Togo 228
• Tokelau 690
• Tonga 676
• Trinidad & Tobago 1
• Tunisia 216
• Turkey 90
• Turkmenistan 993
• Turks & Caicos Islands 1
• Tuvalu 688
• U.S. Virgin Islands 1
• Uganda 256
• Ukraine 380
• United Arab Emirates 971
• United Kingdom 44
• United States 1
• Uruguay 598
• Uzbekistan 998
• Vanuatu 678
• Vatican City 39
• Venezuela 58
• Vietnam 84
• Wallis & Futuna 681
• Western Sahara 212
• Yemen 967
• Zambia 260
• Zimbabwe 263
  
  

Resume/CV*

Attach

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Full Legal Name*

Full Legal Name in Native Language (e.g. Chinese Characters, Cyrillic, Farsi etc)*

Website

LinkedIn Profile

Google Scholar

If you have a Google Scholar page, please provide its URL.

GitHub Profile

X Profile

Your Location*

If working in the US, will you now, or in the future, require sponsorship for employment visa status (e.g., H-1B visa) to legally work in the US?*

Select...

Have you worked with us before?*

Select...

What exceptional work have you done?*

In 100 words or less, tell us about a piece of work you are most proud of.

What makes you the ideal candidate for this position?

How did you hear about us?*

Select...

Powered by

Greenhouse