Member of Technical Staff, Frontiers of Deep Learning Scaling
xAI(1 month ago)
About this role
The role sits on xAI’s Pretrain team focused on discovering new scaling paradigms for model intelligence and effective compute. The position involves exploring what to scale (data, architectures, learning paradigms) and how to scale (large-scale, long-duration stable training runs) to push next-level model capabilities. Candidates will join a small, hands-on engineering organization with a flat structure and high expectations for initiative and execution. The role is based in the Bay Area and contributes directly to xAI’s mission to advance AI understanding of the universe.
Required Skills
- Python
- JAX
- PyTorch
- Rust
- Data Cleaning
- Continual Learning
- Model Architecture
- Attention
- Non-Autoregressive
- Training Stability
+6 more
About xAI
x.aiCloudflare is an internet performance and security company that operates a global edge network to accelerate and protect websites, APIs, and applications with services like CDN, DDoS mitigation, DNS, and a web application firewall. It also provides edge computing (Cloudflare Workers), Zero Trust access and gateway tools, object storage (R2), streaming, load balancing, and other edge-first products so compute and data run closer to users. Developers and organizations pick Cloudflare for its large Anycast network, integrated security controls, simple APIs, and pricing that scales from free developer tiers to enterprise offerings.
Apply instantly with AI
Let ApplyBlast auto-apply to jobs like this for you. Save hours on applications and land your dream job faster.
More jobs at xAI
Similar Jobs
Member of Technical Staff (GPU Engineer)
Reka(26 days ago)
Member of Technical Staff, Machine Learning Datasets
Runway (1 year ago)
Senior Machine Learning Engineer (Large Systems)
Graphcore(7 days ago)
Member of Technical Staff - Diffusion Model
Moonlake AI(2 months ago)
Staff Research Engineer, Pre-training Science
Reddit(8 days ago)
Software Engineer, AI Products
DeepMind(2 months ago)