#!/usr/bin/env python3
import torch
import safetensors
import transformers

print(f"torch version: {torch.__version__}")
print(f"safetensors version: {safetensors.__version__}")
print(f"transformers version: {transformers.__version__}")
print(f"CUDA available: {torch.cuda.is_available()}")

x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])
print(f"x + y = {x + y}")