Thanks to visit codestin.com
Credit goes to Github.com

Skip to content

Tags: AlibabaPAI/xla

Tags

v2.3.1

Toggle v2.3.1's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode 
Support SPMD fsdp compute dtype (#13)

* feat: support compute dtype in spmd fsdp

* fix: import functions from python fsdp

* fix: unused import

* fix: remove comment

v2.3.0

Toggle v2.3.0's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode 
Support low cpu mem usage in SPMD-FSDP (#11)

* feat: support low cpu mem usage in spmd-fsdp

* fix linting