Sieve let’s you run code on GPUs. To do so, all you need to do is add a gpu tag to your function with the GPU you want. Today, Sieve offers a couple machine configurations listed below.

NameGPUMemory (GB)vCPUsTag
T4 (default)T4164t4
A100-40GBA100-40GB8512a100
A100-20GBA100-20GB42.56a100-20gb
V100V100164v100
L4L4328l4

You can specify one of these in code as follows.

@sieve.function(
	name="gpu_function",
	gpu="t4" # Tag
)
def runs_on_gpu(video: sieve.Video) -> sieve.Video:
	...

Read more about the way this field works in SDK reference.