The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning. Learn more →
Top 23 pytorch-implementation Open-Source Projects
-
InfluxDB
Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
-
pytorch-seq2seq
Tutorials on implementing a few sequence-to-sequence (seq2seq) models with PyTorch and TorchText.
-
machine-learning-articles
🧠💬 Articles I wrote about machine learning, archived from MachineCurve.com.
-
pytorch-GAT
My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
-
SimCLR
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations (by sthalles)
-
medicaldetectiontoolkit
The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
-
pytorch-image-classification
Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.
-
6DRepNet
Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.
-
text-to-motion
Official implementation for "Generating Diverse and Natural 3D Human Motions from Texts (CVPR2022)."
-
explainable-cnn
📦 PyTorch based visualization package for generating layer-wise explanations for CNNs.
-
pytorch-learn-reinforcement-learning
A collection of various RL algorithms like policy gradients, DQN and PPO. The goal of this repo will be to make it a go-to resource for learning about RL. How to visualize, debug and solve RL problems. I've additionally included playground.py for learning more about OpenAI gym, etc.
-
Data-Efficient-Reinforcement-Learning-with-Probabilistic-Model-Predictive-Control
Unofficial Implementation of the paper "Data-Efficient Reinforcement Learning with Probabilistic Model Predictive Control", applied to gym environments
-
SaaSHub
SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
Project mention: Building an AI Game Bot 🤖Using Imitation Learning and 3D Convolution ResNet | dev.to | 2024-01-02def compute_mean_std(dataloader): ''' We assume that the images of the dataloader have the same height and width source: https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/Basics/pytorch_std_mean.py ''' # var[X] = E[X**2] - E[X]**2 channels_sum, channels_sqrd_sum, num_batches = 0, 0, 0 for batch_images, labels in tqdm(dataloader): # (B,H,W,C) batch_images = batch_images.permute(0,3,4,2,1) channels_sum += torch.mean(batch_images, dim=[0, 1, 2, 3]) channels_sqrd_sum += torch.mean(batch_images ** 2, dim=[0, 1, 2,3]) num_batches += 1 mean = channels_sum / num_batches std = (channels_sqrd_sum / num_batches - mean ** 2) ** 0.5 return mean, std compute_mean_std(dataloader)
Project mention: [DISC] - The angel who came to pick me up is a Gal (Oneshot by Shiraishi Kouhei) | /r/manga | 2023-09-06OCR works pretty good. ocr.space, ocr.best and cotrans.touhou.ai/ are all pretty nice.
Project mention: [P] 2D Gaussian Splatting a great starting point for people who want to delve deeper | /r/MachineLearning | 2023-10-19Github : https://github.com/OutofAi/2D-Gaussian-Splatting
Project mention: Face recognition models require different thresholds for different races? [D] | /r/MachineLearning | 2023-05-26Hey, first of all, don't use FaceNet. It is outdated and performs way worse than modern solutions. You can use Elastic Face or MagFace (or any other SOTA FR model).
pytorch-implementation related posts
- Building an AI Game Bot 🤖Using Imitation Learning and 3D Convolution ResNet
- What can be the reasons of BatchNorm working and Dropout not working in YoloV1 Pytorch implementation?
- [D] Is it possible to train the same LLM instance on different users' data?
- [P] minLoRA: An Easy-to-Use PyTorch Library for Applying LoRA to PyTorch Models
- How to create a custom parallel corpus for machine translation with recent versions of pytorch and torchtext?
- Help with VAE
- I need help knowing how to improve a CycleGan I am working on.
-
A note from our sponsor - WorkOS
workos.com | 29 Apr 2024
Index
What are some of the best open-source pytorch-implementation projects? This list will help you:
Project | Stars | |
---|---|---|
1 | Machine-Learning-Collection | 6,936 |
2 | PyTorch-VAE | 5,989 |
3 | pytorch-seq2seq | 5,169 |
4 | manga-image-translator | 4,206 |
5 | ALAE | 3,494 |
6 | machine-learning-articles | 3,093 |
7 | pytorch-GAT | 2,222 |
8 | SimCLR | 2,117 |
9 | medicaldetectiontoolkit | 1,266 |
10 | pytorch-image-classification | 899 |
11 | 6DRepNet | 499 |
12 | First-steps-towards-Deep-Learning | 436 |
13 | text-to-motion | 393 |
14 | minLoRA | 387 |
15 | ConvLSTM-PyTorch | 350 |
16 | 2D-Gaussian-Splatting | 289 |
17 | explainable-cnn | 212 |
18 | gan-vae-pretrained-pytorch | 162 |
19 | ElasticFace | 152 |
20 | pytorch-learn-reinforcement-learning | 139 |
21 | Data-Efficient-Reinforcement-Learning-with-Probabilistic-Model-Predictive-Control | 114 |
22 | torch-metrics | 109 |
23 | Pix2Seq | 109 |
Sponsored