PyTorch Ecosystem
PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.
Core Projects
MorePopular Projects
More
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
haystack
AI orchestration framework to build customizable, production-ready LLM applications
pytorch-handbook
pytorch handbook是一本开源的书籍，目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门，其中包含的Pytorch教程全部通过测试保证可以成功运行
DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective
ncnn
ncnn is a high-performance neural network inference framework optimized for the mobile platform
datasets
🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools
pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes
Up and Coming Projects
More
ML-Nexus
ML Nexus is an open-source collection of machine learning projects, covering topics like neural networks, computer vision, and NLP
Gated-Slot-Attention
An implementation of the paper: "Gated Slot Attention for Efficient Linear-Time Sequence Modeling" in PyTorch
Intel_Image_Classification
The solution involves using computer vision and a custom PyTorch-based object detection network to identify different types of apparel from images, with an API for predictions, dockerized and deployed on AWS
mnist-digit-classification
building a machine learning model to classify handwritten digits using the MNIST dataset
Federated-Learning-Simulation-1GPU-MI-IS
Federated Learning Simulation on a Single GPU with Model Interpretability and Interactive Visualization
multihead_attention_implementation
Implementation of Multihead attention mechanism using numpy and pyTorch
acehub
AI Champions for Excellence: Fresh, informative courses and content designed to help developers, researchers, and leaders advance in the field of AI
Tutorial-OpenCV-on-JetsonNano-with-CUDA-support
Installation of OpenCV on JetsonNano with CUDA support
faas-dl-inference
My work towards better understanding how the FaaS paradigm can be adapted for Edge-Cloud environments for improving distributed, containerized ML inferencing workflows
llama.js
run LLMs (llama, mamba, nemo, mistral) at native speeds from Javascript, Typescript
wcmlbook
Python code for the book 'Wireless Communications and Machine Learning' by Le Liang, Shi Jin, and Geoffrey Ye Li
micrograd.zig
small (and partial) autograd engine inspired from Karpathy's micrograd and PyTorch, written in Zig
Sapiens-Pytorch-Inference
Minimal code and examnples for inferencing Sapiens foundation human models in Pytorch
NLP-Projects
- I share in this repository my NLP personal projects that I am working on mainly in NLP
