ML Research Scientist (probabilistic inference)

LawZero is a non-profit organization committed to advancing research and creating technical solutions that enable safe-by-design 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