av transformers numpy torch