xla

f2d83a3a - Allow einsum to take optional parameter to align with torch (#4040)

Allow einsum to take optional parameter to align with torch (#4040) * Allow einsum to take optional parameter to align with torch * See what's wrong with lint * Fix lint * Add torch pin * Removing pin for landing