pytorch

ca41c5f0 - [functorch] Added tests for vmap and autocast (pytorch/functorch#327)

[functorch] Added tests for vmap and autocast (pytorch/functorch#327) Description: - Added tests for vmap and autocast on CUDA/CPU - Tests are failing somehow on CPU - Removed unused imports in test_vmap.py Related to pytorch/functorch#305