Amog Kamsetty c8b112ec46 [Train] Support amp for models with a custom __getstate__ method (#25335) ... The current implementation of amp does not work if the model that is being wrapped defines a custom __getstate__ method. It would fail at the assertion like here: https://discuss.ray.io/t/ray-train-hangs-for-long-time/6333/7. This PR fixes amp for this case, and adds tests for it.		2022-06-02 18:13:13 -07:00
..
ray	[Train] Support amp for models with a custom __getstate__ method (#25335)	2022-06-02 18:13:13 -07:00
requirements	Remove MLDataset (#25041)	2022-05-25 09:33:29 -07:00
asv.conf.json	[docs] Move all /latest links to /master (#11897)	2020-11-10 10:53:28 -08:00
build-wheel-macos-arm64.sh	[python3.10] build python310 wheels (#24829)	2022-05-16 12:36:33 -07:00
build-wheel-macos.sh	[python3.10] build python310 wheels (#24829)	2022-05-16 12:36:33 -07:00
build-wheel-manylinux2014.sh	[python3.10] build python310 wheels (#24829)	2022-05-16 12:36:33 -07:00
build-wheel-windows.sh	[python3.10] build python310 wheels (#24829)	2022-05-16 12:36:33 -07:00
MANIFEST.in	Includes .pyi files in package data. (#21247)	2021-12-27 11:50:02 -08:00
README-building-wheels.md	[build] Build wheels with manylinux2014 (#11621)	2020-11-03 19:36:32 -08:00
requirements.txt	[core] Basic end-2-end multi-node tests for GCS HA in CI. (#25114)	2022-06-02 02:41:38 +00:00
requirements_linters.txt	Remove yapf dependency (#23656)	2022-04-04 21:50:04 -07:00
requirements_ml_docker.txt	[AIR] Add distributed torch_geometric example (#23580)	2022-04-21 09:48:43 -07:00
setup.py	[CI] Bump Bazel version to 4.2.2 (#24242)	2022-05-26 17:09:40 -07:00