ML Research Scientist (probabilistic inference)

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

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