Member of Technical Staff - Thermodynamic ML

Extropic is seeking a Member of Technical Staff to join their ML team, focusing on developing and deploying probabilistic machine learning models in a thermodynamic paradigm. The role involves research, experimentation, implementation, and deployment of models, collaborating with a multidisciplinary team of researchers and engineers. Ideal candidates will have a strong background in probability, linear algebra, and deep learning theory, with experience in training and deploying high-performance models.

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, including 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)