Google Git
Sign in
opensecura / 3p / openxla / iree / 2311e0458bfe47a9f9a681c5e1da5dc0dedd7129 / . / samples / colab
tree: 7f5b8438ae185ab86c71b556f8755226ac867266 [path history] [tgz]
  1. low_level_invoke_function.ipynb
  2. pytorch_aot_advanced.ipynb
  3. pytorch_aot_simple.ipynb
  4. pytorch_huggingface_whisper.ipynb
  5. pytorch_jit.ipynb
  6. README.md
  7. tensorflow_edge_detection.ipynb
  8. tensorflow_hub_import.ipynb
  9. tensorflow_mnist_training.ipynb
  10. tensorflow_resnet.ipynb
  11. test_notebooks.py
  12. tflite_text_classification.ipynb
samples/colab/README.md

Google Colaboratory (Colab) Notebooks

These Colab notebooks contain interactive sample applications using IREE's Python bindings and ML framework integrations.

Notebooks

FrameworkNotebook fileDescriptionLink
Genericlow_level_invoke_functionShows low level IREE python binding conceptsOpen In Colab
PyTorchpytorch_aot_advancedUses Turbine to export a PyTorch programOpen In Colab
PyTorchpytorch_aot_simpleUses Turbine to export a simple PyTorch programOpen In Colab
PyTorchpytorch_jitUses Turbine for eager execution in a PyTorch sessionOpen In Colab
PyTorchpytorch_huggingface_whisperImports the Whisper model from Hugging FaceOpen In Colab
TensorFlowtensorflow_edge_detectionPerforms image edge detectionOpen In Colab
TensorFlowtensorflow_hub_importImports a MobileNet V2 model from TensorFlow HubOpen In Colab
TensorFlowtensorflow_mnist_trainingCompiles, trains, and executes a neural networkOpen In Colab
TensorFlowtensorflow_resnetRuns a pretrained ResNet50 modelOpen In Colab
TFLitetflite_text_classificationRuns a pretrained text classification modelOpen In Colab

Working with GitHub

Refer to Colab's GitHub demo for general information about using Colab with GitHub.

To make changes to a notebook in this repository, one possible workflow is:

  • Open or create the notebook in Colab
  • Connect to a hosted or local runtime
  • Make your changes, run the notebook, etc.
  • Download the modified notebook using File > Download .ipynb
  • Move the downloaded notebook file into a clone of this repository and submit a pull request

Testing

This notebooks are tested continuously by the samples.yml CI job.