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/NVIDIA/retinanet-examples
Fast and accurate object detection with end-to-end GPU optimization
deep-learning neural-network object-detection python pytorch retinanet tensorrt
Last synced: 07 Aug 2024
https://github.com/jacobgil/pytorch-pruning
PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
Last synced: 01 Aug 2024
https://github.com/baguasys/bagua
Bagua Speeds up PyTorch
bagua deep-learning distributed distributed-computing distributed-systems machine-learning pytorch rust-lang
Last synced: 24 Sep 2024
https://github.com/BaguaSys/bagua
Bagua Speeds up PyTorch
bagua deep-learning distributed distributed-computing distributed-systems machine-learning pytorch rust-lang
Last synced: 31 Jul 2024
https://github.com/WenjieDu/PyPOTS
A Python toolkit/library for reality-centric machine/deep learning and data mining on partially-observed time series, including SOTA neural network models for scientific analysis tasks of imputation, classification, clustering, forecasting, & anomaly detection on incomplete industrial (irregularly-sampled) multivariate TS with NaN missing values
classification clustering data-mining data-science deep-learning forecasting healthcare imputation incomplete industrial interpolation machine-learning missing-values missingness neural-network partially-observed-time-series pytorch science-research time-series time-series-analysis
Last synced: 01 Aug 2024
https://github.com/moemen95/Pytorch-Project-Template
A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models
Last synced: 03 Aug 2024
https://github.com/THUDM/SwissArmyTransformer
SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.
pretrained-models pytorch transformer
Last synced: 01 Aug 2024
https://github.com/switchablenorms/Switchable-Normalization
Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10779
convolutional-neural-networks deeplearning imagenet normalization pytorch
Last synced: 02 Aug 2024
https://github.com/VikParuchuri/zero_to_gpt
Go from no deep learning knowledge to implementing GPT.
deep-learning python pytorch tutorial
Last synced: 01 Aug 2024
https://github.com/moemen95/PyTorch-Project-Template
A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models
Last synced: 01 Aug 2024
https://github.com/explosion/curated-transformers
🤖 A PyTorch library of curated Transformer models and their composable components
albert bert camembert dolly2 falcon gptneox llama llm llms nlp pytorch roberta transformer transformers xlm-roberta
Last synced: 24 Sep 2024
https://github.com/opendilab/lightzero
[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios
alpha-beta-pruning alphazero atari board-game board-games continuous-control efficientzero gomoku gumbel-muzero gym mcts mcts-algorithm monte-carlo-tree-search muzero pytorch reinforcement-learning sampled-muzero self-play stochastic-muzero tictactoe
Last synced: 25 Sep 2024
https://github.com/NirAharon/BOT-SORT
BoT-SORT: Robust Associations Multi-Pedestrian Tracking
multi-object-tracker multi-object-tracking multi-pedestrian-tracking pytorch tracking-by-detection yolov7 yolox
Last synced: 02 Aug 2024
https://github.com/amusi/PyTorch-From-Zero-To-One
PyTorch从入门到精通
deep-learning pytorch pytorch-tutorials
Last synced: 31 Jul 2024
https://github.com/ZhiqingXiao/rl-book
Source codes for the book "Reinforcement Learning: Theory and Python Implementation"
deep-reinforcement-learning gym openai-gym python pytorch reinforcement-learning tensorflow tensorflow2
Last synced: 01 Aug 2024
https://github.com/frotms/paddleocr2pytorch
PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)
crnn db multilingual-models ocr paddleocr paddlepaddle pytorch pytorchocr
Last synced: 25 Sep 2024
https://github.com/gmalivenko/pytorch2keras
PyTorch to Keras model convertor
converter darts deep-learning deep-learning-models densenet imagenet keras keras-model-convertor keras-models keras-neural-networks keras-tensorflow machinelearning models-converter onnx pytorch pytorch-models resnet tensorflow tensorflow-models tensorflowjs
Last synced: 24 Sep 2024
https://github.com/nerox8664/pytorch2keras
PyTorch to Keras model convertor
converter darts deep-learning deep-learning-models densenet imagenet keras keras-model-convertor keras-models keras-neural-networks keras-tensorflow machinelearning models-converter onnx pytorch pytorch-models resnet tensorflow tensorflow-models tensorflowjs
Last synced: 04 Aug 2024
https://github.com/mlmed/torchxrayvision
TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.
chest-radiographs chest-xray chest-xray-images cxr cxr-images dataset deep-learning foundation-models image-classification machine-learning medical medical-ai medical-application medical-image-analysis medical-image-processing medical-imaging pytorch torchxrayvision transfer-learning
Last synced: 03 Aug 2024
https://github.com/philipperemy/n-beats
Keras/Pytorch implementation of N-BEATS: Neural basis expansion analysis for interpretable time series forecasting.
deep-learning neural-networks pytorch series-forecasting
Last synced: 03 Aug 2024
https://github.com/lonePatient/Bert-Multi-Label-Text-Classification
This repo contains a PyTorch implementation of a pretrained BERT model for multi-label text classification.
albert bert fine-tuning multi-label-classification nlp pytorch pytorch-implmention text-classification transformers xlnet
Last synced: 04 Aug 2024
https://github.com/vahidk/tfrecord
Standalone TFRecord reader/writer with PyTorch data loaders
dataset loader pytorch tensorflow tfrecord
Last synced: 03 Aug 2024
https://github.com/airctic/icevision
An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
ai annotation-parsers coco-dataset coco-parser computer-vision deep-learning effecientdet fastai faster-rcnn mask-rcnn object-detection pycocotools python pytorch pytorch-lightning tutorials voc-dataset voc-parser
Last synced: 03 Aug 2024
https://github.com/microsoft/GODEL
Large-scale pretrained models for goal-directed dialog
conversational-ai data-processing dialogpt dialogue dialogue-systems grounded-generation language-grounding language-model machine-learning pretrained-model pytorch text-data text-generation transformer transformers
Last synced: 31 Jul 2024
https://github.com/wasidennis/AdaptSegNet
Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)
adversarial-learning computer-vision deep-learning domain-adaptation generative-adversarial-network pytorch semantic-segmentation
Last synced: 05 Aug 2024
https://github.com/perone/medicaltorch
A medical imaging framework for Pytorch
computer-vision deep-learning machine-learning medical-imaging python pytorch
Last synced: 03 Aug 2024
https://github.com/zisianw/FaceBoxes.PyTorch
A PyTorch Implementation of FaceBoxes
face-detection faceboxes pytorch
Last synced: 31 Jul 2024
https://github.com/NVIDIA/modulus
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
deep-learning machine-learning nvidia-gpu physics pytorch
Last synced: 31 Jul 2024
https://github.com/ypwhs/dl-engineer-guidebook
深度学习工程师生存指南
deep-learning deeplearning gitbook guidebook hardware homebrew installation-notes linux macos pytorch tensorflow
Last synced: 02 Aug 2024
https://github.com/easezyc/deep-transfer-learning
A collection of implementations of deep domain adaptation algorithms
deep-transfer-learning domain-adaptation pytorch transfer-learning
Last synced: 02 Aug 2024
https://github.com/pnnl/neuromancer
Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.
constrained-optimization control-systems deep-learning differentiable-control differentiable-optimization differentiable-programming dynamical-systems nonlinear-dynamics nonlinear-optimization physics-informed-ml pytorch
Last synced: 01 Aug 2024
https://github.com/allegro/allRank
allRank is a framework for training learning-to-rank neural models based on PyTorch.
click-model deep-learning information-retrieval learning-to-rank machine-learning ndcg python pytorch ranking transformer
Last synced: 02 Aug 2024
https://github.com/RL-VIG/LibFewShot
LibFewShot: A Comprehensive Library for Few-shot Learning. TPAMI 2023.
few-shot-learning fine-tuning image-classification meta-learning pytorch
Last synced: 03 Aug 2024
https://github.com/baal-org/baal
Bayesian active learning library for research and industrial usecases.
active-learning ai bayesian-active-learning deep-learning machine-learning python pytorch
Last synced: 03 Aug 2024
https://github.com/BUPT-GAMMA/OpenHGNN
This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.
dgl graph-neural-networks heterogeneous pytorch
Last synced: 08 Aug 2024
https://github.com/nyukat/breast_cancer_classifier
Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening
breast-cancer breast-cancer-diagnosis classification deep-learning medical-image-analysis medical-imaging neural-network pretrained-models pytorch tensorflow
Last synced: 04 Aug 2024
https://github.com/yzhangcs/parser
:rocket: State-of-the-art parsers for natural language.
constituency-parsing dependency-parsing pytorch semantic-dependency-parsing semiring state-of-the-art structured-prediction transformers
Last synced: 01 Aug 2024
https://github.com/pytorch/translate
Translate - a PyTorch Language Library
artificial-intelligence machine-learning onnx pytorch
Last synced: 26 Sep 2024
https://github.com/smilelight/lightNLP
基于Pytorch和torchtext的自然语言处理深度学习框架。
cbow chinese cws deep-learning dependency-parsing hierarchical-softmax language-model nature-language-process negative-sampling nlp pos python pytorch relation-extraction sentence-similarity skip-gram srl text-classification torchtext word2vec
Last synced: 03 Aug 2024
https://github.com/voldemortX/pytorch-auto-drive
PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help
apex cityscapes culane deeplab erfnet fcn gtav lane-detection lstr mobilenet onnx pascal-voc pytorch resa scnn semantic-segmentation tensorboard tensorrt tusimple
Last synced: 31 Jul 2024
https://github.com/NVlabs/ODISE
Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]
deep-learning diffusion-models instance-segmentation open-vocabulary open-vocabulary-segmentation open-vocabulary-semantic-segmentation open-world-classification open-world-object-detection panoptic-segmentation pytorch semantic-segmentation text-image-retrieval zero-shot-learning
Last synced: 31 Jul 2024
https://github.com/bamos/densenet.pytorch
A PyTorch implementation of DenseNet.
deep-learning densenet pytorch
Last synced: 31 Jul 2024
https://github.com/nashory/pggan-pytorch
:fire::fire: PyTorch implementation of "Progressive growing of GANs (PGGAN)" :fire::fire:
celeba-hq-dataset gan generative-adversarial-network progressive-gan progressively-growing-gan pytorch tensorboard
Last synced: 01 Aug 2024
https://github.com/RedisAI/RedisAI
A Redis module for serving tensors and executing deep learning graphs
machine-learning onnxruntime pytorch redisai serving-tensors tensorflow
Last synced: 31 Jul 2024
https://github.com/diegoalejogm/gans
Generative Adversarial Networks implemented in PyTorch and Tensorflow
cyclegan dcgans gans gans-collections paper-implementations pytorch tensorflow
Last synced: 07 Aug 2024
https://github.com/geoopt/geoopt
Riemannian Adaptive Optimization Methods with pytorch optim
optimization pytorch riemannian-geometry riemannian-manifold riemannian-optimization
Last synced: 31 Jul 2024
https://github.com/MontaEllis/Pytorch-Medical-Segmentation
This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.
medical-image-processing pytorch segmentation torchio
Last synced: 07 Aug 2024
https://github.com/Denys88/rl_games
RL implementations
deep-learning pytorch reinforcement-learning
Last synced: 01 Aug 2024
https://github.com/wujiyang/Face_Pytorch
face recognition algorithms in pytorch framework, including arcface, cosface, sphereface and so on
arcface cosface face-recognition pytorch sphereface
Last synced: 31 Jul 2024
https://github.com/Replicable-MARL/MARLlib
One repository is all that is necessary for Multi-agent Reinforcement Learning (MARL)
deep-reinforcement-learning multi-agent-reinforcement-learning pytorch ray rllib
Last synced: 31 Jul 2024
https://github.com/kimhc6028/relational-networks
Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)
deep-learning pytorch relational-networks
Last synced: 01 Aug 2024
https://github.com/MinZHANG-WHU/Change-Detection-Review
A review of change detection methods, including codes and open data sets for deep learning. From paper: change detection based on artificial intelligence: state-of-the-art and challenges.
artificial-intelligence caffe change-detection code deep-learning keras machine-learning matlab open-datasets python pytorch remote-sensing review sar streetview tensorflow unsupervised-learning
Last synced: 01 Aug 2024
https://github.com/NirAharon/BoT-SORT
BoT-SORT: Robust Associations Multi-Pedestrian Tracking
multi-object-tracker multi-object-tracking multi-pedestrian-tracking pytorch tracking-by-detection yolov7 yolox
Last synced: 31 Jul 2024
https://github.com/bayesiains/nflows
Normalizing flows in PyTorch
density-estimation generative-model normalizing-flows pytorch
Last synced: 30 Jul 2024
https://github.com/mert-kurttutan/torchview
torchview: visualize pytorch models
deep-learning keras machine-learning neural-network python pytorch torchviz visualization
Last synced: 25 Sep 2024
https://github.com/mimoralea/gdrl
Grokking Deep Reinforcement Learning
algorithms artificial-intelligence deep-learning deep-reinforcement-learning docker gpu machine-learning neural-networks numpy numpy-tutorial nvidia-docker pytorch pytorch-tutorials reinforcement-learning
Last synced: 25 Sep 2024
https://github.com/agrimgupta92/sgan
Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018
deep-learning generative-adversarial-network human-trajectory-prediction pytorch social-navigation trajectory-prediction
Last synced: 02 Aug 2024
https://github.com/jingweiz/pytorch-rl
Deep Reinforcement Learning with pytorch & visdom
a3c acer actor-critic deep-learning deep-reinforcement-learning dqn pytorch pytorch-a3c reinforcement-learning trpo visdom
Last synced: 07 Aug 2024
https://github.com/megvii-research/IJCAI2023-CoNR
IJCAI2023 - Collaborative Neural Rendering using Anime Character Sheets
anime computer-graphics computer-vision deep-learning pytorch
Last synced: 01 Aug 2024
https://github.com/dvgodoy/PyTorchStepByStep
Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"
cnn-pytorch deep-learning python pytorch pytorch-tutorial rnn-pytorch
Last synced: 31 Jul 2024
https://github.com/kakaobrain/torchgpipe
A GPipe implementation in PyTorch
checkpointing deep-learning gpipe model-parallelism parallelism pipeline-parallelism pytorch
Last synced: 01 Aug 2024
https://github.com/polarisZhao/PFLD-pytorch
PFLD pytorch Implementation
face-landmark-detection ncnn pfld pfld-ncnn pfld-pytorch pytorch
Last synced: 31 Jul 2024
https://github.com/996refuse/zheye
者也 - 知乎 倒立的文字 汉字验证码识别程序
gmm neural-network python pytorch zhihu
Last synced: 03 Aug 2024
https://github.com/xxxnell/how-do-vits-work
(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
loss-landscape pytorch self-attention transformer vision-transformer
Last synced: 03 Aug 2024
https://github.com/havakv/pycox
Survival analysis with PyTorch
deep-learning machine-learning neural-networks python pytorch survival-analysis
Last synced: 02 Aug 2024
https://github.com/kazuto1011/grad-cam-pytorch
PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)
deconvnet grad-cam guided-backpropagation occlusion-sensitivity pytorch visualization
Last synced: 03 Aug 2024
https://github.com/MushroomRL/mushroom-rl
Python library for Reinforcement Learning.
atari ddpg deep-learning deep-reinforcement-learning dqn mujoco openai-gym pybullet pytorch qlearning reinforcement-learning rl sac trpo
Last synced: 01 Aug 2024
https://github.com/pranz24/pytorch-soft-actor-critic
PyTorch implementation of soft actor critic
deep-reinforcement-learning pytorch pytorch-implmention reinforcement-learning soft-actor-critic
Last synced: 02 Aug 2024
https://github.com/alibaba/BladeDISC
BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
compiler deep-learning inference-optimization machine-learning mlir neural-network pytorch tensorflow
Last synced: 31 Jul 2024
https://github.com/torrvision/crayon
A language-agnostic interface to TensorBoard
crayon data-visualization deep-learning docker pytorch tensorboard tensorflow torch7
Last synced: 03 Aug 2024
https://github.com/rahulschand/gpu_poor
Calculate token/s & GPU memory requirement for any LLM. Supports llama.cpp/ggml/bnb/QLoRA quantization
ggml gpu huggingface language-model llama llama2 llamacpp llm pytorch quantization
Last synced: 27 Sep 2024
https://github.com/benedekrozemberczki/ClusterGCN
A PyTorch implementation of "Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks" (KDD 2019).
clustering community-detection deep-learning deepwalk diff2vec gcn gemsec graph-clustering graph-convolution graph-convolutional-networks graph-neural-networks graph2vec graphsage louvain metis musae neural-network node-classification node2vec pytorch
Last synced: 02 Aug 2024
https://github.com/pytorch/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Last synced: 24 Sep 2024
https://github.com/MediaBrain-SJTU/MING
明医 (MING):中文医疗问诊大模型
consultation huggingface llm medical pytorch transformers
Last synced: 03 Aug 2024
https://github.com/jeya-maria-jose/Medical-Transformer
Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021
attention deep-learning medical-image-analysis medical-imaging pytorch segmentation transformer transformers
Last synced: 03 Aug 2024
https://github.com/parrt/tensor-sensor
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
debugging deep-learning jax matrix numpy python pytorch tensorflow tracing vector
Last synced: 25 Sep 2024
https://github.com/youngwanLEE/centermask2
[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation
anchor-free centermask cvpr2020 detectron2 instance-segmentation object-detection pytorch real-time vovnet vovnetv2
Last synced: 30 Jul 2024
https://github.com/kwea123/VTuber_Unity
Use Unity 3D character and Python deep learning algorithms to stream as a VTuber!
dlib face-detection face-landmark-detection face-pose gaze-estimation python pytorch unity unity-chan vrm vtuber
Last synced: 02 Aug 2024
https://github.com/vandit15/Class-balanced-loss-pytorch
Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"
computer-vision cvpr2019 deep-learning loss-functions pytorch
Last synced: 03 Aug 2024
https://github.com/opentensor/bittensor
Internet-scale Neural Networks
ai blockchain cryptocurrency deep-learning machine-learning neural-networks p2p p2p-network polkadot pytorch substrate torch
Last synced: 02 Aug 2024
https://github.com/kaituoxu/Speech-Transformer
A PyTorch implementation of Speech Transformer, an End-to-End ASR with Transformer network on Mandarin Chinese.
asr attention attention-is-all-you-need end-to-end pytorch self-attention transformer
Last synced: 07 Aug 2024
https://github.com/denis2054/transformers-for-nlp-2nd-edition
Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more
bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax
Last synced: 02 Aug 2024
https://github.com/mit-han-lab/tinyengine
[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning; [NeurIPS 2022] MCUNetV3: On-Device Training Under 256KB Memory
c codegenerator cpp deep-learning edge-computing microcontroller neural-architecture-search pytorch quantization tinyml
Last synced: 31 Jul 2024
https://github.com/kuan-wang/pytorch-mobilenet-v3
MobileNetV3 in pytorch and ImageNet pretrained models
classification imagenet mobilenet mobilenetv2 mobilenetv3 pytorch
Last synced: 31 Jul 2024
https://github.com/higgsfield/Capsule-Network-Tutorial
Pytorch easy-to-follow Capsule Network tutorial
capsule-network clean-code easy-to-use pytorch pytorch-tutorials
Last synced: 31 Jul 2024
https://github.com/spijkervet/simclr
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
contrastive-learning pytorch representation-learning simclr unsupervised-learning
Last synced: 27 Sep 2024
https://github.com/berniwal/swin-transformer-pytorch
Implementation of the Swin Transformer in PyTorch.
artificial-intelligence attention-model deep-learning machine-learning pytorch transformer-architecture transformer-pytorch
Last synced: 07 Aug 2024
https://github.com/SimonVandenhende/Multi-Task-Learning-PyTorch
PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
computer-vision eccv2020 multi-task-learning nyud pascal pytorch scene-understanding segmentation
Last synced: 02 Aug 2024
https://github.com/rusty1s/pytorch_cluster
PyTorch Extension Library of Optimized Graph Cluster Algorithms
cluster-algorithms geometric-deep-learning graph-neural-networks pytorch
Last synced: 01 Aug 2024
https://github.com/teddykoker/torchsort
Fast, differentiable sorting and ranking in PyTorch
cuda-kernel pytorch ranking sort
Last synced: 03 Aug 2024
https://github.com/positive666/yolo_research
based on yolo-high-level project (detect\pose\classify\segment\):include yolov5\yolov7\yolov8\ core ,improvement research ,SwintransformV2 and Attention Series. training skills, business customization, engineering deployment C
attention-mechanism custom-networ deepstream object-detection pytorch swintransformer yolov5 yolov7 yolov8
Last synced: 02 Aug 2024
https://github.com/benedekrozemberczki/simgnn
A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
attention-mechanism deep-learning gcn ged gnn graph-attention graph-classification graph-convolution graph-edit-distance graph-embedding graph-similarity machine-learning network-embedding neural-network pytorch simgnn sklearn tensor-network tensorflow wsdm
Last synced: 26 Sep 2024
https://github.com/G-U-N/PyCIL
PyCIL: A Python Toolbox for Class-Incremental Learning
continual-learning deep-learning incremental-learning lifelong-learning machine-learning pytorch reproducible-research
Last synced: 03 Aug 2024
https://github.com/hengyuan-hu/bottom-up-attention-vqa
An efficient PyTorch implementation of the winning entry of the 2017 VQA Challenge.
bottom-up-attention pytorch vqa
Last synced: 01 Aug 2024
https://github.com/wmcnally/kapao
KAPAO is an efficient single-stage human pose estimation model that detects keypoints and poses as objects and fuses the detections to predict human poses.
deep-learning human-pose-estimation pose-estimation pytorch yolo
Last synced: 02 Aug 2024
https://github.com/fastnlp/fastHan
fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具,像spacy一样调用方便。
bert cws fastnlp joint-model ner parser pos python pytorch
Last synced: 31 Jul 2024
https://github.com/SongweiGe/rich-text-to-image
Rich-Text-to-Image Generation
computer-vision diffusion-models pytorch rich-text text-to-image-generation
Last synced: 31 Jul 2024
https://github.com/asyml/texar-pytorch
Integrating the Best of TF into PyTorch, for Machine Learning, Natural Language Processing, and Text Generation. This is part of the CASL project: http://casl-project.ai/
bert casl-project data-processing deep-learning dialog-systems gpt-2 machine-learning machine-translation natural-language-processing python pytorch roberta texar texar-pytorch text-data text-generation xlnet
Last synced: 02 Aug 2024
https://github.com/fregu856/deeplabv3
PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.
autonomous-driving computer-vision deep-learning machine-learning pytorch semantic-segmentation
Last synced: 07 Aug 2024
https://github.com/shibing624/similarities
Similarities: a toolkit for similarity calculation and semantic search. 相似度计算、匹配搜索工具包,支持亿级数据文搜文、文搜图、图搜图,python3开发,开箱即用。
bm25 deep-learning faiss image-search image-similarity matching nlp pytorch search-engine similarity similarity-search text-matching
Last synced: 25 Sep 2024