Research Scientist - Machine Learning

Extropic is seeking a Research Scientist to develop probabilistic ML theory, scale model up experiments, and deploy performant models. The role involves collaborating with researchers and engineers to derive models, optimize performance, and create production-ready models for domain experts. This is a full-time position in a hybrid work environment.

Requirements

• Experience in scientific Python and at least one deep learning framework (PyTorch, JAX, TensorFlow, Keras)
• Extremely strong foundations in probability and linear algebra
• Familiarity with deep learning theory and literature (theory of over-parameterization and scaling laws)
• Publications in top ML conferences (NeurIPS, ICML, ICLR, CVPR)
• Experience training high-performance models, including familiarity with infrastructure (Slurm, Ray, Weights & Biases)
• Experience deploying models, including familiarity with infrastructure (Ray, AWS, ONNX)
• Strong theoretical background in information geometry
• Strong theoretical background in random matrix theory
• Strong grasp of computational Bayesian methods, including MCMC sampling methods and variational inference