Nikolaos daydreamt

daydreamt / BDK.py

Last active March 26, 2021 12:51

Bayesian Dark Knowledge: A modification of https://github.com/google/jax/blob/master/examples/mnist_classifier.py implementing https://arxiv.org/abs/1506.04416

	import jax
	import jax.numpy as np
	from jax.experimental import stax
	from jax.experimental import optimizers
	from jax.experimental.stax import Dense, Relu, Tanh, Softmax, LogSoftmax
	from jax import jit, grad, random

	import time
	import itertools

daydreamt / funs.txt

Last active April 23, 2016 10:02

Experiments in wrapping torch from OCaml.