Low-latency neural decoders for linear and non-linear block codes