ML Research Scientist (probabilistic inference)

LawZero is a non-profit organization committed to advancing research and creating technical solutions for safe-by-design AI systems. The ML Research Scientist will develop and evaluate probabilistic inference methods, translate theoretical insights into practical implementations, and collaborate on research involving probabilistic models and AI safety.

Responsibilities:

• Develop amortized inference methods suitable for high-dimensional discrete and continuous distributions
• Develop parameter- and structure-learning methods for large probabilistic graphical models that benefit from amortized probabilistic inference
• Design evaluation strategies for methods that rely on probabilistic inference
• Collaborate with mathematicians on theory related to learning and inference in probabilistic models
• Translate theoretical proposals into high quality implementations in a programming language such as Python
• Analyze and interpret experimental results to steer future research directions
• Communicate complex findings effectively to various stakeholders