https://github.com/ShawonAshraf/annotated-jax

Deep Learning examples using the Jax ecosystem of libraries
https://github.com/ShawonAshraf/annotated-jax

computer-vision deep-learning flax jax nlp notebook-jupyter tutorial

Last synced: 6 months ago
JSON representation

Deep Learning examples using the Jax ecosystem of libraries

• Host: GitHub
• URL: https://github.com/ShawonAshraf/annotated-jax
• Owner: ShawonAshraf
• License: mit
• Created: 2022-01-04T01:02:21.000Z (about 4 years ago)
• Default Branch: main
• Last Pushed: 2024-10-27T18:38:25.000Z (over 1 year ago)
• Last Synced: 2024-11-14T02:23:44.214Z (about 1 year ago)
• Topics: computer-vision, deep-learning, flax, jax, nlp, notebook-jupyter, tutorial
• Language: Jupyter Notebook
• Homepage:
• Size: 9.4 MB
• Stars: 4
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# annotated-jax

> Deep Learning examples using the Jax ecosystem of libraries

[![Build Status](https://shawonashraf.visualstudio.com/jax-examples/_apis/build/status%2FShawonAshraf.jax_examples?branchName=main)](https://shawonashraf.visualstudio.com/jax-examples/_build/latest?definitionId=13&branchName=main)

## env setup

> [!IMPORTANT]

> Ensure that you've `uv` installed: [link](https://docs.astral.sh/uv/getting-started/installation/). 

```bash

uv sync

source .venv/bin/activate

```

> [!TIP]

> You can also use the provided devcontainer configuration.

> ```bash

> devcontainer up --workspace-folder .

> ```

## run jupyter lab

```bash

# inside the project root

jupyter lab

```