Parallelism works. NaNs in sgd though.