Skip to content

Commit

Permalink
Add eager mnist example.
Browse files Browse the repository at this point in the history
  • Loading branch information
stellaraccident committed Oct 10, 2023
1 parent cc4ab94 commit 8b8aab2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ and not everything works at head with release binaries at present.
* [AOT MLP with a dynamic batch size](https://github.com/nod-ai/SHARK-Turbine/blob/main/examples/aot_mlp/mlp_export_dynamic.py)
* [AOT llama2](https://github.com/nod-ai/SHARK-Turbine/blob/main/examples/llama2_inference/llama2.ipynb):
Dynamic sequence length custom compiled module with state management internal to the model.
* [Eager MNIST with `toch.compile`](https://github.com/nod-ai/SHARK-Turbine/blob/main/examples/eager_mlp/mlp_eager_simple.py)

## Developers

Expand Down

0 comments on commit 8b8aab2

Please sign in to comment.