Dev container is outdated

[jorisSchaller]

Provide as much information as possible. At least, this should include a description of your issue and steps to reproduce the problem. If possible also provide a summary of what steps or workarounds you have already tried.

System information

• OS Platform and Distribution: MacOS
• Flax, jax, jaxlib versions (obtain with pip show flax jax jaxlib: Not applicable, problems occurs when building the dev containers.
• Python version: 3.12
• GPU/TPU model and memory: Not Applicable
• CUDA version (if applicable): Not Applicable
• Docker desktop version: 4.28.0
• Docker engine version: 25.0.3
• VS code version: 1.94.2

Problem you have encountered:

Unable to build the devcontainer.

What you expected to happen:

I expected to connect to the dev container after building it, but the instruction at ./dev/README.md aren't up to date. Specifically the configuration files.

Logs, error messages, etc:

1. ERROR: Ignored the following versions that require a different python version ERROR: Could not find a version that satisfies the requirement jaxlib (from versions: none)
2. The 'sklearn' PyPI package is deprecated, use 'scikit-learn'
3. file:///workspaces/flax/flax does not appear to be a Python project: neither 'setup.py' nor 'pyproject.toml' found.

Steps to reproduce:

Whenever possible, please provide a minimal example. Please consider submitting it as a Colab link.

1. Copy ./dev/.devcontainer to the root directory of the repo.
2. Open root directory in VS code.
3. Launch the container when prompted by VS Code.
4. Check that it fails to run.