accelerate safetensors av numpy pillow einops