switch freqs to linspace for rotary embedding

fix layernorm

allow for input channels greater than 3, for doing conditioning in DD…

allow for rectangular images, as well as tailoring hyperparameters fo…

add axial rotary positional embedding

allow for conditioning on time for DDPM training

make attention window-able

first pass of uformer, without relative positional embedding and wind…