GitHub

Flexible and powerful tensor operations for readable and reliable code.

Previous updates - flax and oneflow backend added - torch.jit.script is supported for pytorch layers - powerful EinMix added to einops. [Einmix tutorial notebook ...