Lorentz group equivariant autoencoders