Create and drive engineering agendas to advance multimodal capabilities, with emphasis on image and video generation, editing, understanding, controllable/long-horizon synthesis, agentic planning, RL training, and world simulation (including audio integration for richer video…
Develop reliable, performant, and maintainable software responsible for training Grok models in our world-class compute clusters. Provide technical direction for future training architectures and hardware designs.
Design, build, and optimize massive GPU clusters for extreme-scale training and inference workloads. Develop and tune low-level CUDA kernels (GeMM, Attention, etc.), using CUTLASS, Tensor Cores, and Nsight for maximum performance.
Architect and implement scalable distributed infrastructure for model serving (load balancing, auto-scaling, batch scheduling, global KV cache). Optimize latency and throughput of model inference under real production workloads.
Own and drive features from inception and design to implementation and launch, being the web expert on your team.
Develop backend services, APIs, and data models to support high-volume, multi-user environments. Work with iOS, Android & Web client engineers to ship products.
Collaborate with engineering, product leadership, business leadership, and engineers across the company. Maintain and improve the systems that serve our existing customers while designing and shipping new products that expand what the platform can do.
Design and build large scale search infrastructure (real-time indexing, serving, and data pipelines) that handles massive write and query loads with strict latency and reliability requirements.
Design, build, and launch to production new features and improvements aimed at unifying common components across the storage systems.
Dive into the low-level stack to profile, debug, and optimize performance across diverse systems, including GPUs, Linux kernel, networking, and filesystems, to achieve peak efficiency.
Showing 11–20 of 94 jobs
Show more jobs
Palo Alto, Canada
USA
London, GB
Dublin, IE