Amortized optimization for optimal transport