generalise map to functions of arbitrary shape (reshape version) - Githubissues

joelberkeley / spidr

Accelerated machine learning with dependent types

Apache License 2.0

76 stars 4 forks source link

generalise map to functions of arbitrary shape (reshape version) #288

Closed joelberkeley closed 2 years ago

joelberkeley commented 2 years ago

xla::While is slow on GPU so we'll try a different way