AI Research Engineer, Scaling jobs in United States
cer-icon
Apply on Employer Site
company-logo

1X · 4 months ago

AI Research Engineer, Scaling

1X is a company that builds humanoid robots to address labor shortages and create abundance. The AI Research Engineer, Scaling will design and build infrastructure for large-scale training and deployment of robots, focusing on optimizing performance and efficiency in both datacenter and edge environments.

Artificial Intelligence (AI)ManufacturingRobotics
check
H1B Sponsor Likelynote

Responsibilities

Own and lead scaling of distributed training and inference systems
Ensure compute resources are optimized to make data the primary constraint
Enable massive training runs (1000+ GPUs) using robot data, with robust fault tolerance, experiment tracking, and distributed operations
Optimize inference throughput for datacenter use cases such as world models and diffusion engines
Reduce latency and enhance performance for on-device robot policies using techniques such as quantization, scheduling, and distillation

Qualification

PythonC++Distributed training frameworksOptimizing inference performanceCUDATriton kernelsQuantization strategiesExperiment managementMulti-node debuggingHigh scaling mindset

Required

Strong programming experience in Python and/or C++
Deep intuitive understanding of training and inference speed bottlenecks and scaling laws
A mindset aligned with extremely high scaling: belief that scale is foundational to enabling humanoid robotics
Degree in Computer Science or a related field
Experience with distributed training frameworks (e.g., TorchTitan, DeepSpeed, FSDP/ZeRO), multi-node debugging, and experiment management
Proven skills in optimizing inference performance using graph compilers, batching/scheduling, and serving systems like TensorRT or equivalents
Familiarity with quantization strategies (PTQ, QAT, INT8/FP8) and tools such as TensorRT and bitsandbytes
Experience developing or tuning CUDA or Triton kernels with understanding of hardware-level optimization (vectorization, tensor cores, memory hierarchies)

Benefits

Health, dental, and vision insurance
401(k) with company match
Paid time off and holidays

Company

1X

twittertwittertwitter
company-logo
1X is an engineering and robotics company that develops humanoid robots designed to work alongside humans.

H1B Sponsorship

1X has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (16)
2024 (3)
2023 (1)

Funding

Current Stage
Growth Stage
Total Funding
$136.49M
Key Investors
EQT VenturesOpenAI Startup Fund
2025-07-07Series B
2024-01-11Series B· $99.94M
2024-01-11Secondary Market
Company data provided by crunchbase