Homework 08: Detecting Distribution Shift on MNIST using Bayesian Neural Networks

import torch
import torch.nn as nn
import numpy as np
import matplotlib.pyplot as plt
from tqdm.auto import trange, tqdm

torch.manual_seed(42)
np.random.seed(42)

from PIL import Image
from torchvision import datasets
from torch.nn.functional import softmax
from torchvision.transforms.functional import rotate

Acknowledgments

• Initial version: Mark Neubauer

© Copyright 2025