Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
PyTorch
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-07-29 13:55:27 UTC
- JSON Representation
https://github.com/pemami4911/neural-combinatorial-rl-pytorch
PyTorch implementation of Neural Combinatorial Optimization with Reinforcement Learning https://arxiv.org/abs/1611.09940
neural-combinatorial-optimization pytorch reinforcement-learning seq2seq
Last synced: 01 Aug 2024
https://github.com/cvxgrp/pymde
Minimum-distortion embedding with PyTorch
cuda dimensionality-reduction embedding feature-vectors gpu graph-embedding machine-learning pytorch visualization
Last synced: 02 Aug 2024
https://github.com/VITA-Group/FasterSeg
[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang
cityscapes latency neural-architecture-search pytorch semantic-segmentation tensorrt
Last synced: 07 Aug 2024
https://github.com/dair-ai/pytorch_notebooks
🔥 A collection of PyTorch notebooks for learning and practicing deep learning
deep-learning machine-learning nlp pytorch
Last synced: 03 Sep 2024
https://github.com/rowanz/neural-motifs
Code for Neural Motifs: Scene Graph Parsing with Global Context (CVPR 2018)
pytorch scene-graph vision visual-genome
Last synced: 01 Aug 2024
https://github.com/kmeng01/rome
Locating and editing factual associations in GPT (NeurIPS 2022)
gpt interpretability pytorch transformers
Last synced: 31 Jul 2024
https://github.com/OlaWod/FreeVC
FreeVC: Towards High-Quality Text-Free One-Shot Voice Conversion
pytorch speech voice-conversion
Last synced: 03 Aug 2024
https://github.com/610265158/Peppa_Pig_Face_Landmark
A simple face detect and alignment method, which is easy and stable.
face-alignment face-keypoint face-mask facedetect keypoint pytorch
Last synced: 02 Aug 2024
https://github.com/HoloClean/holoclean
A Machine Learning System for Data Enrichment.
data-enrichment data-science inference-engine machine-learning pytorch
Last synced: 02 Aug 2024
https://github.com/bradyz/cross_view_transformers
Cross-view Transformers for real-time Map-view Semantic Segmentation (CVPR 2022 Oral)
cvpr2022 deep-learning pytorch transformer
Last synced: 31 Jul 2024
https://github.com/wenet-e2e/wespeaker
Research and Production Oriented Speaker Verification, Recognition and Diarization Toolkit
asv campplus cnceleb dino ecapa-tdnn eres2net nist-sre plda production-ready pytorch repvgg resnet self-supervised-learning speaker-diarization speaker-recognition speaker-verification ssl tdnn voxceleb xvector
Last synced: 08 Aug 2024
https://github.com/qinzheng93/GeoTransformer
[CVPR2022] Geometric Transformer for Fast and Robust Point Cloud Registration
attention-mechanism computer-vision deep-learning neural-network point-cloud point-cloud-registration pytorch registration transformer
Last synced: 31 Jul 2024
https://github.com/metaopt/torchopt
TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
automatic-differentiation bilevel-optimization deep-learning differentiable-optimization differentiable-programming functional-programming implicit-differentiation meta-learning meta-reinforcement-learning meta-rl optimization optimizer pytorch
Last synced: 02 Aug 2024
https://github.com/allenai/allennlp-models
Officially supported AllenNLP models
Last synced: 01 Aug 2024
https://github.com/torchmd/torchmd
End-To-End Molecular Dynamics (MD) Engine using PyTorch
coarse-grained-molecular-dynamics machine-learning molecular-dynamics neural-networks pytorch pytorch-lightning
Last synced: 05 Aug 2024
https://github.com/Oldpan/Faceswap-Deepfake-Pytorch
Faceswap with Pytorch or DeepFake with Pytorch
Last synced: 01 Aug 2024
https://github.com/WangQvQ/YOLOMagic
YOLO Magic🪄 is an extension based on Ultralytics' YOLOv5, designed to provide more powerful functionality and simpler operations for visual tasks.
deep-learning gradio machine-learning onnx pytorch tflite yolo yolov5
Last synced: 02 Aug 2024
https://github.com/xiaoyufenfei/LEDNet
LEDNet: A Lightweight Encoder-Decoder Network for Real-time Semantic Segmentation
cityscape-dataset computer-vision lednet pytorch real-time semantic-segmentation
Last synced: 03 Aug 2024
https://github.com/ViTAE-Transformer/ViTDet
Unofficial implementation for [ECCV'22] "Exploring Plain Vision Transformer Backbones for Object Detection"
deep-learning object-detection pytorch vision-transformer
Last synced: 03 Aug 2024
https://github.com/Audio-WestlakeU/FullSubNet
PyTorch implementation of "FullSubNet: A Full-Band and Sub-Band Fusion Model for Real-Time Single-Channel Speech Enhancement."
audio band denoising full-band narrow-band noise-reduction paper pretrained-model pytorch reproducible-research single-channel speech speech-enhancement speech-processing speech-separation sub-band
Last synced: 05 Aug 2024
https://github.com/DagnyT/hardnet
Hardnet descriptor model - "Working hard to know your neighbor's margins: Local descriptor learning loss"
computer-vision convolutional-neural-networks deep-learning image-matching image-retrieval metric-learning nips-2017 pytorch
Last synced: 01 Aug 2024
https://github.com/THUwangcy/ReChorus
“Chorus” of recommendation models: a light and flexible PyTorch framework for Top-K recommendation.
framework pytorch ranking recommendation
Last synced: 08 Aug 2024
https://github.com/allenai/tango
Organize your experiments into discrete steps that can be cached and reused throughout the lifetime of your research project.
ai machine-learning nlp python python3 pytorch
Last synced: 02 Aug 2024
https://github.com/understandable-machine-intelligence-lab/Quantus
Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations
deep-learning explainable-ai interpretability machine-learning pytorch quantification-evaluation-methods reproducibility tensorflow xai
Last synced: 03 Aug 2024
https://github.com/prlz77/ResNeXt.pytorch
Reproduces ResNet-V3 with pytorch
Last synced: 31 Jul 2024
https://github.com/hfslyc/AdvSemiSeg
Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018
adversarial-learning computer-vision deep-learning pytorch semantic-segmentation semi-supervised-learning
Last synced: 31 Jul 2024
https://github.com/SimonBlanke/Hyperactive
An optimization and data collection toolbox for convenient and fast prototyping of computationally expensive models.
automated-machine-learning bayesian-optimization data-science deep-learning feature-engineering hyperactive hyperparameter-optimization keras machine-learning model-selection neural-architecture-search optimization parallel-computing parameter-tuning python pytorch scikit-learn xgboost
Last synced: 01 Aug 2024
https://github.com/medipixel/rl_algorithms
Structural implementation of RL key algorithms
deep-learning dqn gym policy-gradient python3 pytorch reinforcement-learning
Last synced: 02 Aug 2024
https://github.com/flashinfer-ai/flashinfer
FlashInfer: Kernel Library for LLM Serving
cuda flash-attention gpu large-large-models llm-inference pytorch tvm
Last synced: 04 Aug 2024
https://github.com/NTMC-Community/MatchZoo-py
Facilitating the design, comparison and sharing of deep text matching models.
deep-learning matching natural-language-processing neural-network pytorch text text-matching
Last synced: 03 Aug 2024
https://github.com/aimagelab/mammoth
An Extendible (General) Continual Learning Framework based on Pytorch - official codebase of Dark Experience for General Continual Learning
continual-learning dark-experience-replay deep-learning der experience-replay knowledge-distillation neurips2020 pytorch
Last synced: 03 Aug 2024
https://github.com/cardwing/Codes-for-PVKD
Point-to-Voxel Knowledge Distillation for LiDAR Semantic Segmentation (CVPR 2022)
cvpr2022 lidar-segmentation nuscenes point-cloud-segmentation pytorch range-image semantickitti waymo-dataset
Last synced: 31 Jul 2024
https://github.com/yihongXU/deepMOT
Official Implementation of How To Train Your Deep Multi-Object Tracker (CVPR2020)
deep-learning multi-object-tracking python pytorch
Last synced: 02 Aug 2024
https://github.com/PennyLaneAI/qml
Introductions to key concepts in quantum programming, as well as tutorials and implementations from cutting-edge quantum computing research.
autograd automatic-differentiation demo key-concepts neural-networks pytorch qml quantum-chemistry quantum-computing quantum-machine-learning tensorflow tutorials
Last synced: 01 Aug 2024
https://github.com/ldeecke/gmm-torch
Gaussian mixture models in PyTorch.
gaussian-mixture-models pytorch
Last synced: 05 Aug 2024
https://github.com/shayneobrien/generative-models
Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN
autoencoder began discriminator dragan fgan fishergan gan generative-adversarial-network generative-models infogan lsgan machine-learning mmgan nsgan python pytorch ragan vae wasserstein wgan
Last synced: 01 Aug 2024
https://github.com/kaushalshetty/Structured-Self-Attention
A Structured Self-attentive Sentence Embedding
attention attention-mechanism attention-model attention-weights classification deep-learning python3 pytorch self-attention self-attentive-rnn sentence-embeddings visualization
Last synced: 05 Aug 2024
https://github.com/ksw0306/FloWaveNet
A Pytorch implementation of "FloWaveNet: A Generative Flow for Raw Audio"
clarinet generative-flow glow pytorch wavenet
Last synced: 03 Aug 2024
https://github.com/polyaxon/traceml
Engine for ML/Data tracking, visualization, explainability, drift detection, and dashboards for Polyaxon.
dask data-exploration data-profiling data-quality data-quality-checks data-science data-visualization dataframes dataops explainable-ai matplotlib mlops pandas pandas-summary plotly pytorch spark statistics tensorflow tracking
Last synced: 01 Aug 2024
https://github.com/hamelsmu/code_search
Code For Medium Article: "How To Create Natural Language Semantic Search for Arbitrary Objects With Deep Learning"
code-search data-science deep-learning fastai keras machine-learning machine-learning-on-source-code ml-on-code natural-language-processing nlp python pytorch search search-algorithm searching-algorithms semantic-search semantic-search-engine tensorflow tutorial
Last synced: 17 Aug 2024
https://github.com/facebookresearch/NeuralCompression
A collection of tools for neural compression enthusiasts.
compression deep-learning jax machine-learning neural-compression python pytorch
Last synced: 05 Aug 2024
https://github.com/pyg-team/pytorch-frame
Tabular Deep Learning Library for PyTorch
data-frame deep-learning pytorch tabular-learning
Last synced: 01 Aug 2024
https://github.com/salesforce/matchbox
Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.
deep-learning minibatch nlp pytorch
Last synced: 03 Aug 2024
https://github.com/dongminlee94/deep_rl
PyTorch implementation of deep reinforcement learning algorithms
a2c ddpg ddqn deep-reinforcement-learning dqn model-free-rl npg ppo pytorch sac sac-aea td3 trpo vpg
Last synced: 02 Aug 2024
https://github.com/amirassov/kaggle-imaterialist
The First Place Solution of Kaggle iMaterialist (Fashion) 2019 at FGVC6
computer-vision deep-learning hybrid-task-cascade imaterialist instance-segmentation kaggle kaggle-imaterialist mmdetection object-detection pytorch
Last synced: 31 Jul 2024
https://github.com/swz30/CycleISP
[CVPR 2020--Oral] CycleISP: Real Image Restoration via Improved Data Synthesis
camera-imaging-pipeline computer-vision cvpr2020 cycleisp data-synthesis image-denoising image-restoration low-level-vision pytorch raw2rgb rgb2raw
Last synced: 01 Aug 2024
https://github.com/AnubhavGupta3377/Text-Classification-Models-Pytorch
Implementation of State-of-the-art Text Classification Models in Pytorch
attention classification convolutional-neural-networks deep-learning fasttext nlp pytorch rcnn recurrent-neural-networks seq2seq transformer
Last synced: 02 Aug 2024
https://github.com/giddyyupp/ganilla
Official Pytorch implementation of GANILLA
deep-learning gan gans generative-adversarial-network illustration illustration-drawing illustrations illustrator image-to-image-translation pytorch
Last synced: 07 Aug 2024
https://github.com/mahmoodlab/HIPT
Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)
computational-pathology cvpr cvpr2022 deep-learning hierarchical-attention-networks high-resolution histopathology pretrained-weights pytorch self-supervised-learning transfer-learning unsupervised-learning vision-transformer weakly-supervised-learning
Last synced: 02 Aug 2024
https://github.com/RUCAIBox/CRSLab
CRSLab is an open-source toolkit for building Conversational Recommender System (CRS).
conversation-system conversational-recommendation deep-learning dialog-system graph-neural-network human-machine-interaction knowledge-graph pretrained-models pytorch recommendation recommender-system text-generation
Last synced: 03 Aug 2024
https://github.com/adambielski/CapsNet-pytorch
PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules
capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch
Last synced: 07 Aug 2024
https://github.com/jaketae/storyteller
Multimodal AI Story Teller, built with Stable Diffusion, GPT, and neural text-to-speech
ddpm diffusion-models gpt image-generation natural-language-generation pytorch stable-diffusion text-to-image text-to-speech text-to-video video-generation
Last synced: 04 Aug 2024
https://github.com/locuslab/optnet
OptNet: Differentiable Optimization as a Layer in Neural Networks
deep-learning machine-learning optimization paper pytorch
Last synced: 01 Aug 2024
https://github.com/vikasverma1077/manifold_mixup
Code for reproducing Manifold Mixup results (ICML 2019)
data-augumentation deep-learning deep-neural-networks icml2019 pytorch regularization supervised-learning supervised-machine-learning
Last synced: 01 Aug 2024
https://github.com/cvqluu/Angular-Penalty-Softmax-Losses-Pytorch
Angular penalty loss functions in Pytorch (ArcFace, SphereFace, Additive Margin, CosFace)
am-softmax arcface embedding face-recognition face-verification fashion-mnist fmnist-dataset loss-function loss-functions metric-learning normface pytorch speaker-recognition sphereface
Last synced: 01 Aug 2024
https://github.com/ma-xu/pointMLP-pytorch
[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"
modelnet40 pointcloud pytorch scanobjectnn
Last synced: 31 Jul 2024
https://github.com/EagleW/PaperRobot
Code for PaperRobot: Incremental Draft Generation of Scientific Ideas
attention-mechanism datasets end-to-end-learning generation memory-networks natural-language-generation nlp paper-generation pytorch text-generation
Last synced: 04 Aug 2024
https://github.com/archinetai/surgeon-pytorch
A library to inspect and extract intermediate layers of PyTorch models.
artificial-intelligence deep-learning pytorch
Last synced: 03 Aug 2024
https://github.com/v-iashin/video_features
Extract video features from raw videos using multiple GPUs. We support RAFT flow frames as well as S3D, I3D, R(2+1)D, VGGish, CLIP, and TIMM models.
audio-features clip feature-extraction i3d ig65m laion multi-gpu optical-flow parallel pytorch r2plus1d raft resnet s3d swin timm vggish video-features visual-features vit
Last synced: 01 Aug 2024
https://github.com/kevinzakka/recurrent-visual-attention
A PyTorch Implementation of "Recurrent Models of Visual Attention"
attention pytorch ram recurrent-attention-model recurrent-models
Last synced: 31 Jul 2024
https://github.com/crowsonkb/style-transfer-pytorch
Neural style transfer in PyTorch.
neural-art pytorch style-transfer
Last synced: 05 Aug 2024
https://github.com/backtime92/CRAFT-Reimplementation
CRAFT-Pyotorch:Character Region Awareness for Text Detection Reimplementation for Pytorch
Last synced: 01 Aug 2024
https://github.com/fkodom/fft-conv-pytorch
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
convolution image-processing neural-networks python3 pytorch
Last synced: 05 Aug 2024
https://github.com/wildltr/ptranking
Learning to Rank in PyTorch
learning-to-rank pytorch ranking
Last synced: 02 Aug 2024
https://github.com/POSTECH-CVLab/SCNeRF
[ICCV21] Self-Calibrating Neural Radiance Fields
calibration computer-vision deep-learning implicit-representions nerf pytorch self-calibration
Last synced: 08 Aug 2024
https://github.com/microsoft/archai
Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.
automated-machine-learning automl darts deep-learning hyperparameter-optimization machine-learning model-compression nas neural-architecture-search petridish python pytorch
Last synced: 01 Aug 2024
https://github.com/criteo-research/reco-gym
Code for reco-gym: A Reinforcement Learning Environment for the problem of Product Recommendation in Online Advertising
criteo openai-gym pytorch recommendation-system
Last synced: 03 Aug 2024
https://github.com/JingyibySUTsoftware/Yolov5-deepsort-driverDistracted-driving-behavior-detection
基于深度学习的驾驶员分心驾驶行为(疲劳+危险行为)预警系统使用YOLOv5+Deepsort实现驾驶员的危险驾驶行为的预警监测
deeplearning deepsort detection dlib dlib-face-detection dlib-tracker driver-behavior object-detection opencv playsound pyqt5 python pytorch sort tired tracker yolov5 yolov5-deepsort
Last synced: 02 Aug 2024
https://github.com/chingyaoc/ggnn.pytorch
A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN)
Last synced: 31 Jul 2024
https://github.com/Ha0Tang/SelectionGAN
[CVPR 2019 Oral] Multi-Channel Attention Selection GAN with Cascaded Semantic Guidance for Cross-View Image Translation
adversarial-learning computer-graphics computer-vision cross-view cvpr-2019 cvpr19 cvpr2019 cvusa-dataset dayton deep-learning gans generative-adversarial-network image-generation image-manipulation image-to-image-translation image-translation pytorch semantic-maps
Last synced: 06 Aug 2024
https://github.com/flaport/fdtd
A 3D electromagnetic FDTD simulator written in Python with optional GPU support
3d-fdtd electric-fields fdtd fdtd-simulator magnetic-fields numpy optics photonics physics physics-simulation python pytorch simulation simulation-framework
Last synced: 03 Aug 2024
https://github.com/ZwwWayne/K-Net
[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation
image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation
Last synced: 03 Aug 2024
https://github.com/yinchangchang/ocr_densenet
第一届西安交通大学人工智能实践大赛(2018AI实践大赛--图片文字识别)第一名;仅采用densenet识别图中文字
densenet ocr ocr-recognition python pytorch
Last synced: 01 Aug 2024
https://github.com/Toni-SM/skrl
Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab
deep-learning deepmind gym gymnasium isaac-gym isaac-lab isaac-orbit isaac-sim isaaclab jax machine-learning nvidia-omniverse openai-gym python pytorch reinforcement-learning rl robosuite robotics skrl
Last synced: 01 Aug 2024
https://github.com/EdisonLeeeee/GraphGallery
GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.
adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch
Last synced: 02 Aug 2024
https://github.com/Schlumberger/joint-vae
Pytorch implementation of JointVAE, a framework for disentangling continuous and discrete factors of variation :star2:
disentangled-representations disentanglement generative-models gumbel-softmax pytorch vae
Last synced: 03 Aug 2024
https://github.com/TRI-ML/dd3d
Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.
computer-vision deep-learning pytorch
Last synced: 31 Jul 2024
https://github.com/cure-lab/MagicDrive
[ICLR24] Official implementation of the paper “MagicDrive: Street View Generation with Diverse 3D Geometry Control”
autonomous-vehicles deep-learning diffusion-models image-generation pytorch video-generation
Last synced: 31 Jul 2024
https://github.com/sinkingsugar/nimtorch
PyTorch - Python + Nim
artificial-intelligence artificial-neural-networks cuda machine-learning nim pytorch wasm
Last synced: 31 Jul 2024
https://github.com/Neutone/neutone_sdk
Join the community on Discord for more discussions around Neutone! https://discord.gg/VHSMzb8Wqp
ai audio audio-plugin deep-learning machine-learning music python pytorch realtime-audio sdk vst
Last synced: 01 Aug 2024
https://github.com/cavalleria/cavaface
face recognition training project(pytorch)
amsoftmax apex arcface arcnegface attention-irse circleloss cosface curricularface dataparallel efficientnet face-recognition ghostnet loss mixup model-parallel network pytorch randaugment resnest resnet-irse
Last synced: 31 Jul 2024
https://github.com/davidefiocco/streamlit-fastapi-model-serving
Simple web app example serving a PyTorch model using streamlit and FastAPI
docker-compose fastapi pycones pytorch streamlit
Last synced: 07 Aug 2024
https://github.com/sinAshish/Multi-Scale-Attention
Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"
attention-mechanism biomedical-image-analysis deep-learning encoder-decoder pytorch
Last synced: 03 Aug 2024
https://github.com/clovaai/rexnet
Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models
deep-learning image-classification imagenet object-detection pretrained-model pytorch rexnet transfer-learning
Last synced: 07 Aug 2024
https://github.com/dfdazac/wassdistance
Approximating Wasserstein distances with PyTorch
deep-learning machine-learning optimal-transport pytorch
Last synced: 02 Aug 2024
https://github.com/salesforce/warp-drive
Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)
cuda deep-learning gpu high-throughput multiagent-reinforcement-learning numba pytorch reinforcement-learning
Last synced: 02 Aug 2024
https://github.com/uvipen/Tetris-deep-Q-learning-pytorch
Deep Q-learning for playing tetris game
cv2 deep-q-learning deep-q-network pytorch reinforcement-learning
Last synced: 07 Aug 2024
https://github.com/walktree/libtorch-yolov3
A Libtorch implementation of the YOLO v3 object detection algorithm
Last synced: 02 Aug 2024
https://github.com/ploomber/sklearn-evaluation
Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.
data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow
Last synced: 01 Aug 2024
https://github.com/layumi/AICIty-reID-2020
:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)
aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776
Last synced: 01 Aug 2024
https://github.com/huggingface/large_language_model_training_playbook
An open collection of implementation tips, tricks and resources for training large language models
cuda large-language-models llm nccl nlp performance python pytorch scalability troubleshooting
Last synced: 02 Aug 2024
https://github.com/StepNeverStop/RLs
Reinforcement Learning Algorithms Based on PyTorch
deep-reinforcement-learning gym ml-agents pytorch reinforcement-learning reinforcement-learning-algorithm sac training-agents unity3d
Last synced: 02 Aug 2024
https://github.com/loretoparisi/CapsNet
CapsNet (Capsules Net) in Geoffrey E Hinton paper "Dynamic Routing Between Capsules" - State Of the Art
capsnet capsule-networks chainer deeplearning keras lasagne machine-learning mnist mxnet pytorch tensorflow
Last synced: 01 Aug 2024
https://github.com/sacmehta/ESPNetv2
A light-weight, power efficient, and general purpose convolutional neural network
cnn convolutional-neural-networks deep-learning efficient imagenet lightweight machine-learning pytorch semantic-segmentation
Last synced: 31 Jul 2024
https://github.com/chaitjo/geometric-gnn-dojo
Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.
e3nn equivariance equivariant-networks geometric-deep-learning geometric-graphs graph-isomorphism graph-neural-networks pytorch pytorch-geometric
Last synced: 02 Aug 2024
https://github.com/navneet-nmk/pytorch-rl
This repository contains model-free deep reinforcement learning algorithms implemented in Pytorch
ddpg deep-learning deep-reinforcement-learning dqn experience-replay gan generative-adversarial-network gym mujoco mujoco-py openai openai-gym policy-gradient pytorch reinforcement-learning rl-algorithms robotics vae vae-gan variational-autoencoder
Last synced: 01 Aug 2024
https://github.com/iBelieveCJM/Tricks-of-Semi-supervisedDeepLeanring-Pytorch
PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch
fixmatch ict meanteacher mixmatch pi pseudo-label pytorch semi-supervised-learning ssl tempens vat
Last synced: 03 Aug 2024
https://github.com/THUDM/GraphMAE
GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22
deep-learning graph-neural-networks pytorch self-supervised-learning
Last synced: 08 Aug 2024
https://github.com/nschaetti/EchoTorch
A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.
artificial-intelligence artificial-neural-networks echo-state-networks machine-learning machine-learning-algorithms machinelearning neural-networks python python-toolkit pytorch recurrent-networks recurrent-neural-networks reservoir-computing torch
Last synced: 03 Aug 2024