annotated-jax

Deep Learning examples using the Jax ecosystem of libraries

env setup

Important

Ensure that you've uv installed: link.

uv sync
source .venv/bin/activate

Tip

You can also use the provided devcontainer configuration.

devcontainer up --workspace-folder .

# inside the project root
jupyter lab

Name		Name	Last commit message	Last commit date
Latest commit History 266 Commits
.devcontainer		.devcontainer
.github		.github
basics		basics
checkpoints		checkpoints
images		images
playground		playground
transformer		transformer
utils		utils
.dockerignore		.dockerignore
.gitignore		.gitignore
.python-version		.python-version
LICENSE		LICENSE
README.md		README.md
azure-pipelines.yml		azure-pipelines.yml
ci.Dockerfile		ci.Dockerfile
devcontainer.Dockerfile		devcontainer.Dockerfile
pyproject.toml		pyproject.toml
uv.lock		uv.lock