tinygrad/docs/showcase.md

1.4 KiB

tinygrad Showcase

Despite being a tiny library, tinygrad is capable of doing a lot of things. From state-of-the-art vision to state-of-the-art language models.

Vision

EfficientNet

You can either pass in the URL of a picture to discover what it is:

python3 examples/efficientnet.py https://media.istockphoto.com/photos/hen-picture-id831791190

Or, if you have a camera and OpenCV installed, you can detect what is in front of you:

python3 examples/efficientnet.py webcam

YOLOv8

Take a look at yolov8.py.

yolov8 by tinygrad

Audio

Whisper

Take a look at whisper.py. You need pyaudio and torchaudio installed.

SMALL=1 python3 examples/whisper.py

Generative

Generative Adversarial Networks

Take a look at mnist_gan.py.

mnist gan by tinygrad

Stable Diffusion

python3 examples/stable_diffusion.py

a horse sized cat eating a bagel

"a horse sized cat eating a bagel"

LLaMA

You will need to download and put the weights into the weights/LLaMA directory, which may need to be created.

Then you can have a chat with Stacy:

python3 examples/llama.py