Как импортировать другую ветку пакета (на основе python) в GitHub на локальный компьютер?

У меня есть вопрос об импорте другой ветки репозитория в Google Colab.

Autograd (https://github.com/HIPS/autograd) имеет адаптируемую к частям версию, которая лежит в другой ветке (https://github.com/HIPS/autograd/tree/cupy).I хочу импортировать autograd (удобную для cupy версию в Google Colab и использовать следующую команду:

!pip install git+git://github.com/ericmjl/autograd/tree/cupy

но у него есть следующие ошибки:

Сбор git+git://github.com/ericmjl/autograd/tree/cupy
Клонирование git://github.com/ericmjl/autograd/tree/cupy в /tmp/pip-yoasr9_z-build fatal:

удаленная ошибка: ercmjl/autograd/tree/cupy не является допустимым именем репозитория
Электронная почта [email protected] для получения помощи

Команда «git clone -q git://github.com/ericmjl/autograd/tree/cupy /tmp/pip-yoasr9_z-build» не удалась с кодом ошибки 128 в None

Есть ли кто-нибудь, кто может помочь мне с этим? Огромное спасибо!!