This is the mnist-wasm api