xla
use XLA patched linear in FSDP (fix #3811 and #3718) and expose options on param sharding dim and pinning memory
#3830
Merged
Merged