Merge "Split pip tests in different modules to enable sharding and reduce runtime" into main