diff --git a/README.md b/README.md index 8918c9d..ecfa6f8 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# torchsnapshot +# TorchSnapshot (Beta Release)

build status @@ -8,8 +8,6 @@ bsd license -**This library is currently in Alpha and currently does not have a stable release. The API may change and may not be backward compatible. If you have suggestions for improvements, please open a GitHub issue. We'd love to hear your feedback.** - A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind. @@ -20,9 +18,14 @@ Requires Python >= 3.7 and PyTorch >= 1.12 From pip: ```bash +# Stable +pip install torchsnapshot + +# Nightly pip install --pre torchsnapshot-nightly ``` + From source: ```bash