jax

410dbaee - Merge pull request #37679 from Intel-tensorflow:ashahba/docker-env-improvements