Re: [ot][spam][crazy][data] transformer model 'attention' improvement

This is the transcription of the code I'm first trying with. I haven't tested it yet. I need to generate some data with appropriate dimensions, and I'm somewhat new to jax.numpy .