@Bhack
Sure we can add a non-root user, as long as it works well for both vscode and standalone docker env.
For the nightly image, I don’t think we have a keras-nightly.
We can use a tf-nightly, but I am not sure if it works well with the SSH authentication for GitHub if using codespaces.
Would you help us make this changes? I feel you are more familiar with these setup than me. : )