code base: https://github.com/BlackHC/BatchBALD
