Notebooks
We prepared some notebooks for you, so that you can run directly tutorials in the documentation.
Training
| Notebook | Description | Studio Lab |
|---|---|---|
| Fine-tune BERT for text classification on AWS Trainium | Show how to fine-tune BERT on AWS Trainium for text classification. |  |
Inference
| Notebook | Description | Studio Lab |
|---|---|---|
| Create your own chatbot with llama-2-13B on AWS Inferentia | Show how to run LLama-2 13B chat model on AWS inferentia 2. | |
| How to generate images with Stable Diffusion | Show how to use stable-diffusion v2.1 model to generate images from prompts on Inferentia 2. | |
| How to generate images with Stable Diffusion XL | Show how to use stable-diffusion XL model to generate images from prompts on Inferentia 2. | |
| Compute text embeddings with Sentence Transformers on Inferentia | Show how to use Sentence Transformers to compute sentence / text embeddings on Inferentia 2. | |
| How to compile (if needed) and generate text with CodeLlama 7B | How to use CodeLlama 7B to generate code. Also walks through compilation. | |