<s> import parakeet <EOL> import numpy as np <EOL> @ parakeet . jit <EOL> def matmult ( X , Y ) : <EOL> return np . array ( [ [ np . dot ( x , y ) for y in Y . T ] for x in X ] ) <EOL> n , d = <NUM_LIT> , <NUM_LIT> <EOL> m = <NUM_LIT> <EOL> dtype = '<STR_LIT>' <EOL> X = np . random . randn ( m , d ) . astype ( dtype ) <EOL> Y = np . random . randn ( d , n ) . astype ( dtype ) <EOL> Z = matmult ( X , Y ) </s>
