Neural wave-based imaging with amortized uncertainty quantification