torchwright

A compiler that transforms computation graphs into the weights of a transformer