pytorch/functorch
functorch is JAX-like composable function transforms for PyTorch.
Stars: 1,436Language: Jupyter Notebook
Give AlbumentationsX a star on GitHub — it powers this leaderboard
Star on GitHubfunctorch is JAX-like composable function transforms for PyTorch.