Add joint NLL from Epistemic Neural Networks (https://arxiv.org/abs/2107.08924) to imagenet/* scripts.

[copybara-service[bot]]

Add joint NLL from Epistemic Neural Networks (https://arxiv.org/abs/2107.08924) to imagenet/* scripts.

[google-cla[bot]]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

For more information, open the CLA check for this pull request.