🖥 tinygrad — фреймворк Python для Deep Learning

— pip install tinygrad

tinygrad имеет встроенную библиотеку для работы с нейросетями, используя разные классы, оптимизаторы и инструменты для управления состоянием нейросетей.

В tinygrad есть JIT, чтобы всё работало быстро.
Также tinygrad имеет отличную поддержку нескольких GPU, что позволяет удобно параллелить операции над тензорами с помощью Tensor.shard.

🖥 GitHub
🟡 Доки
🟡 Примеры использования tinygrad

@pythonl