Bat-Sheva Einbinder and Hen Davidov and Yee Whye Teh and Yarin Gal and Yaniv Romano