Add auto_sharding dependency in OSS
Hi @ckstanton This PR is in draft, any update on this? Please. Thank you! > Hi @ckstanton This PR is in draft, any update on this? Please. Thank you!

This was just a PR to demonstrate an issue we're seeing, I wasn't planning on submitting this... sorry!  I'll close this PR now.
Currently we see seg faults when trying to enable auto_sharding in OSS.  This is a minimally-failing example for reproducing the issue.  (This PR only adds the necessary dependency, we do not yet enable the auto_pass in gpu_compiler in OSS.)