Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

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.

https://github.com/goru001/inltk

Natural Language Toolkit for Indic Languages aims to provide out of the box support for various NLP tasks that an application developer might need

data-augmentation deep-learning indic-languages nlp pytorch sentence-embeddings sentence-encoding sentence-similarity word-embeddings

Last synced: 18 Nov 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: 31 Dec 2024

https://github.com/haofanwang/controlnet-for-diffusers

Transfer the ControlNet with any basemodel in diffusers🔥

aigc controlnet diffusers pytorch tutorials

Last synced: 03 Jan 2025

https://github.com/ritchieng/deep-learning-wizard

Open source guides/codes for mastering deep learning to deploying deep learning in production in PyTorch, Python, Apptainer, and more.

apptainer deep-learning llamaindex ollama python3 pytorch

Last synced: 03 Jan 2025

https://github.com/geoopt/geoopt

Riemannian Adaptive Optimization Methods with pytorch optim

optimization pytorch riemannian-geometry riemannian-manifold riemannian-optimization

Last synced: 30 Oct 2024

https://github.com/guochengqian/pointnext

[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies

3d-vision pointcloud pointnet2 pytorch

Last synced: 01 Jan 2025

https://github.com/cloneofsimo/minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.

diffusion pytorch

Last synced: 30 Oct 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: 10 Nov 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: 02 Nov 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: 30 Oct 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: 28 Oct 2024

https://github.com/mosecorg/mosec

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

cv deep-learning gpu hacktoberfest jax llm llm-serving machine-learning machine-learning-platform mlops model-serving mxnet nerual-network python pytorch rust tensorflow tts

Last synced: 03 Jan 2025

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: 15 Nov 2024

https://github.com/mihaidusmanu/d2-net

D2-Net: A Trainable CNN for Joint Description and Detection of Local Features

cnn cvpr2019 local-features pytorch visual-localization

Last synced: 01 Jan 2025

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: 04 Jan 2025

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: 27 Oct 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: 08 Nov 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: 09 Nov 2024

https://github.com/shadow2496/VITON-HD

Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)

deep-learning pytorch

Last synced: 08 Nov 2024

https://github.com/996refuse/zheye

者也 - 知乎 倒立的文字 汉字验证码识别程序

gmm neural-network python pytorch zhihu

Last synced: 01 Jan 2025

https://github.com/microsoft/cameratraps

PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.

camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife

Last synced: 02 Jan 2025

https://github.com/rasbt/LLM-workshop-2024

A 4-hour coding workshop to understand how LLMs are implemented and used

large-language-models llm pytorch

Last synced: 25 Dec 2024

https://github.com/rasbt/llm-workshop-2024

A 4-hour coding workshop to understand how LLMs are implemented and used

large-language-models llm pytorch

Last synced: 03 Jan 2025

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: 15 Nov 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: 04 Jan 2025

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: 15 Nov 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: 04 Jan 2025

https://github.com/oaid/autokernel

AutoKernel 是一个简单易用,低门槛的自动算子优化工具,提高深度学习算法部署效率。

auto autosearch deep-learning halide optimization pytorch reinforcement-learning tengine tensor tensorflow tvm

Last synced: 01 Jan 2025

https://github.com/torrvision/crayon

A language-agnostic interface to TensorBoard

crayon data-visualization deep-learning docker pytorch tensorboard tensorflow torch7

Last synced: 30 Sep 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: 10 Nov 2024

https://github.com/teddykoker/torchsort

Fast, differentiable sorting and ranking in PyTorch

cuda-kernel pytorch ranking sort

Last synced: 15 Nov 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: 15 Nov 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: 26 Oct 2024

https://github.com/ChenRocks/UNITER

Research code for ECCV 2020 paper "UNITER: UNiversal Image-TExt Representation Learning"

pre-training pytorch transformers vision-and-language

Last synced: 28 Nov 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: 04 Jan 2025

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: 27 Nov 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: 02 Jan 2025

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: 03 Jan 2025

https://github.com/tusen-ai/SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

3d-object-detection autonomous-driving pytorch

Last synced: 28 Oct 2024

https://github.com/tusen-ai/sst

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

3d-object-detection autonomous-driving pytorch

Last synced: 01 Jan 2025

https://github.com/microsoft/CameraTraps

PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.

camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife

Last synced: 27 Nov 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: 27 Oct 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: 05 Nov 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 Oct 2024

https://github.com/guanfuchen/semseg

常用的语义分割架构结构综述以及代码复现 华为媒体研究院 图文Caption、OCR识别、图视文多模态理解与生成相关方向工作或实习欢迎咨询 15757172165 https://guanfuchen.github.io/media/hw_zhaopin_20220724_tiny.jpg

bisenet deep-learning deeplab drn enet erfnet fcn frrn future-semantic-segmentation gcn image-segmentation label-refinement-network lrn pytorch segmentation-models segmentation-networks segnet semantic-segmentation shufflenet

Last synced: 21 Nov 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: 28 Oct 2024

https://github.com/hkchengrex/cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation

computer-vision cvpr2024 deep-learning pytorch segmentation video-editing video-object-segmentation video-segmentation

Last synced: 05 Jan 2025

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: 24 Oct 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: 27 Nov 2024

https://github.com/kymatio/kymatio

Wavelet scattering transforms in Python with GPU acceleration

deep-learning keras numpy pytorch scattering-transform scikit-learn tensorflow wavelets

Last synced: 02 Jan 2025

https://github.com/csteinmetz1/auraloss

Collection of audio-focused loss functions in PyTorch

audio loss-functions pytorch

Last synced: 31 Dec 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: 13 Nov 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: 09 Nov 2024

https://github.com/fastnlp/fasthan

fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具,像spacy一样调用方便。

bert cws fastnlp joint-model ner parser pos python pytorch

Last synced: 04 Jan 2025

https://github.com/tencent/patrickstar

PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.

bert gpt nlp pretrained-models pytorch

Last synced: 03 Jan 2025

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: 07 Nov 2024

https://github.com/chenyuntc/pytorch-best-practice

A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats

image-classification pytorch visdom

Last synced: 03 Jan 2025

https://github.com/smilelab-fl/fedlab

A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.

deep-learning federated-learning federated-learning-framework fedlab machine-learning pytorch pytorch-federated-learning

Last synced: 31 Dec 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: 09 Nov 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: 05 Jan 2025

https://github.com/fastnlp/fastHan

fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具,像spacy一样调用方便。

bert cws fastnlp joint-model ner parser pos python pytorch

Last synced: 27 Oct 2024

https://github.com/soapisnotfat/super-resolution

collection of super-resolution models & algorithms

fsrcnn pytorch resolution-models srcnn srgan sup-pixel-cnn

Last synced: 22 Nov 2024

https://github.com/Tencent/PatrickStar

PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.

bert gpt nlp pretrained-models pytorch

Last synced: 16 Nov 2024

https://github.com/microsoft/tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation

mixture-of-experts moe nlp pytorch transformer

Last synced: 03 Jan 2025

https://github.com/paddlepaddle/x2paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)

caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo

Last synced: 26 Dec 2024

https://github.com/guochengqian/PointNeXt

[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies

3d-vision pointcloud pointnet2 pytorch

Last synced: 28 Oct 2024

https://github.com/fboulnois/stable-diffusion-docker

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.

dall-e dalle diffusion docker generative-art huggingface image-generation inpainting midjourney pytorch stable-diffusion tensorflow text-to-image

Last synced: 01 Jan 2025

https://github.com/maximumentropy/seq2seq-pytorch

Sequence to Sequence Models with PyTorch

deep-learning pytorch rnn seq2seq

Last synced: 04 Jan 2025

https://github.com/msnh2012/Msnhnet

🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.

darknet inference-engine jetson-nx mobilenetv2 mobilenetyolo pytorch yolov3 yolov4 yolov5

Last synced: 09 Nov 2024

https://github.com/williamfalcon/test-tube

Python library to easily log experiments and parallelize hyperparameter search for neural networks

caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow

Last synced: 01 Jan 2025

https://github.com/jackroos/VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

bert iclr2020 pre-training pytorch representation-learning self-supervised-learning vision-and-language vl-bert

Last synced: 03 Nov 2024

https://github.com/williamFalcon/test-tube

Python library to easily log experiments and parallelize hyperparameter search for neural networks

caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow

Last synced: 30 Oct 2024

https://github.com/quic/sense

Enhance your application with the ability to see and interact with humans using any RGB camera.

activity-recognition calorie-estimation computer-vision deep-learning fitness-app gesture-recognition neural-networks pytorch video

Last synced: 22 Nov 2024

https://github.com/bobbens/sketch_simplification

Models and code related to sketch simplification of rough sketches.

convolutional-neural-networks deep-learning pytorch siggraph sketch torch

Last synced: 04 Jan 2025

https://github.com/PaddlePaddle/X2Paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)

caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo

Last synced: 06 Nov 2024

https://github.com/allentdan/libtorchtutorials

This is a code repository for pytorch c++ (or libtorch) tutorial.

classification cpp libtorch pytorch pytorch-cpp resnet segmentation tutorial-code unet vgg yolov4 yolov4-tiny

Last synced: 04 Jan 2025

https://github.com/chungyiweng/humannerf

HumanNeRF turns a monocular video of moving people into a 360 free-viewpoint video.

cvpr2022 humannerf nerf pytorch

Last synced: 07 Nov 2024