High-performance machine learning research library with composable transformations of Python functions.