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.
Simple Feed Forward Neural Net with Giga Neurons for classifying GigaChads.
Key Features: (it's our vision for now)
-
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.
-
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.
-
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.
- MLP based Character Level Language Model following Bengio et al. 2003
- CNN for classifying MNIST dataset.
The current recommended way to install GigaTorch is 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
poetry build
poetry run pytest
poetry run black .
poetry run python example/image_models/gigatorch/cnn.py
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
- Add AutoGrad https://pytorch.org/tutorials/beginner/introyt/autogradyt_tutorial.html
- Create Convolution neural-network using GigaTorch.
- Create Recurrent neural-network using GigaTorch.
- Add accelerator support in Tensor class.
Supported accelerators:
- CPU
- CUDA (WIP)
- GPU (OpenCL)
- ANE (Apple Neural Engine)
- Google TPU