Founding Deep Learning Engineer

UniversalAGI is building OpenAI for Physics. AI startup based in San Francisco and backed by Elad Gil, Eric Schmidt, Prith Banerjee, Ion Stoica, Jared Kushner, David Patterson, and Luis Videgaray. We're building foundation AI models for physics that enable end-to-end industrial automation from initial design through optimization, validation, and production.

Requirements

• 3+ years of hands-on experience training deep learning models, with a track record of shipping models to production
• Deep expertise in modern deep learning frameworks (PyTorch, JAX) and model architectures (Transformers, Diffusion Models, Graph Neural Networks, GNNs, CNNs, GCNs, PointNet, RegDGCNN, Neural Operators, etc.)
• Strong foundation in distributed training: Experience with multi-GPU and multi-node training, gradient accumulation, mixed precision, and optimization techniques
• Expert-level Python and proficiency with ML libraries (HuggingFace, PyTorch Lightning, etc.)
• Solid understanding of ML fundamentals: Optimization, regularization, generalization, evaluation metrics, and the full training lifecycle
• Experience with large-scale datasets: Building data pipelines, handling data quality issues, and working with diverse data formats
• Strong intuition for debugging models: Can diagnose training instabilities, convergence issues, and performance bottlenecks
• Research mindset with execution focus: Can read and implement papers quickly, but prioritizes shipping working solutions over perfect ones
• Outstanding problem-solving: Willing to dive deep into unfamiliar domains (physics, CFD, engineering) and learn what's needed
• Excellent communicator: Can explain complex model behavior to customers, engineers, and non-technical stakeholders
• Thrives in ambiguity: Comfortable defining what success looks like and figuring out how to get there

Benefits

• Competitive compensation and equity
• Competitive health, dental, vision benefits paid by the company
• 401(k) plan offering
• Flexible vacation
• Team Building & Fun Activities
• AI tools stipend
• Monthly commute stipend
• Monthly wellness / fitness stipend
• Daily office lunch & dinner covered by the company
• Immigration support