xla

a6ea0f34 - Use tfrt cpu client (#3898)

Use tfrt cpu client (#3898) * Use TFRT CPU client and make parameters modifiable through environment variables. * Minor build fixes and test skeleton * Implement PJRT multi CPU tests * Clarify env var naming and work around for failing test