xla

Add most of the remaining loss, norm and distance functions

#565

Merged