More loop parallelism in transport

Currently we only loop on first dimension