Thank you. I will read the paper. I'm looking for an alternative to pytorch which works great for toy problems but starts to suffer at scale.