Skip to content

orifmilod/GigaTorch

Repository files navigation

GigaTorch

GigaTorch Logo
GigaTorch: Something between PyTorch and TinyGrad. Maintained by GigaChads around the world.


Welcome to GigaTorch, the revolutionary machine learning framework that takes the art of neural network training to unprecedented levels. It is not just any machine learning framework, GigaTorch is a true game-changer that caters specifically to the needs of GigaChads, those exceptional individuals in the field of deep learning.

Screenshot 2023-06-11 at 13 41 13

Simple Feed Forward Neural Net with Giga Neurons for classifying GigaChads.


Key Features: (it's our vision for now)

  1. Unparalleled Performance: GigaTorch takes advantage of cutting-edge optimizations and novel algorithms, enabling lightning-fast model training and inference. With its streamlined execution pipeline, GigaTorch minimizes computational bottlenecks, resulting in significantly reduced training times.

  2. Advanced Model Customization: GigaTorch empowers users to effortlessly design and tailor GigaChad networks to their specific needs. Leverage an extensive library of pre-defined layers, activation functions, and optimization algorithms.

  3. Scalability and Parallelism: GigaTorch seamlessly scales across multiple GPUs and distributed systems, enabling efficient training of large-scale GigaChad models. With built-in support for data parallelism and model parallelism, GigaTorch empowers practitioners to tackle even the most demanding machine learning challenges.

Embrace the power of GigaChad and experience the next generation of machine learning with GigaTorch. Join the GigaTorch community and unlock the true potential of your GigaChad networks. The future of AI begins here.

Current implemented examples

Installation

The current recommended way to install GigaTorch is from source.

From source

git clone https://github.com/orifmilod/GigaTorch.git
cd GigaTorch
# Make sure you have virtualenv installed
pip3 install poetry
poetry config virtualenvs.in-project true
poetry install

Building the project

poetry build

Run the tests

poetry run pytest 

Linting and formating the code in-place

poetry run black .

Running examples

poetry run python example/image_models/gigatorch/cnn.py

Contributing & features to add

If you want to contribute, please take a look at the issues created and project tab, but here is list of features we want to add

Supported accelerators:

  • CPU
  • CUDA (WIP)
  • GPU (OpenCL)
  • ANE (Apple Neural Engine)
  • Google TPU