About the role
Designs and builds AI training platforms for physical AI.
- •Build the next-generation training and learning platform for physical AI at Snowflake.
- •Key Responsibilities Design and build scalable training infrastructure for representation models Develop latent world models that learn environment dynamics Architect and implement action/policy model pipelines Build generative simulator frameworks Develop multimodal generative model capabilities Requirements 8+ years of relevant experience Deep expertise in representation learning, world models, reinforcement learning, generative modeling, robotics/embodied AI, or scientific ML Hands-on experience training large-scale models with distributed compute Strong software engineering fundamentals MS or Ph.D. in Computer Science, Machine Learning, Robotics, Physics, or a related field
Tech stack
PythonTensorFlowPyTorchscikit-learnJAXSnowflake
Match insights
Tech:Python, TensorFlow, PyTorch, scikit-learn, JAX
Level:Lead