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/jiarenchang/psmnet

Pyramid Stereo Matching Network (CVPR2018)

psmnet pytorch stereo-matching stereo-vision

Last synced: 28 Dec 2024

https://github.com/keyu-tian/SparK

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl

Last synced: 28 Oct 2024

https://github.com/fangwei123456/spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

deep-learning dvs machine-learning pytorch snn spiking-neural-networks

Last synced: 24 Dec 2024

https://github.com/speedinghzl/CCNet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/diffeqml/torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

control-theory deep-equilibrium-models deep-learning dynamical-systems implicit-models neural-differential-equations neural-network neural-ode numerical-methods pytorch

Last synced: 28 Dec 2024

https://github.com/nod-ai/shark-studio

SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution

amd apple-silicon deep-learning machine-learning mlir nvidia pytorch

Last synced: 26 Dec 2024

https://github.com/microsoft/mup

maximal update parametrization (µP)

deep-learning machine-learning mup mutransfer python pytorch transformers

Last synced: 24 Dec 2024

https://github.com/goatmessi8/rfbnet

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

detection mobilenet pytorch rfbnet

Last synced: 22 Dec 2024

https://github.com/GOATmessi8/RFBNet

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

detection mobilenet pytorch rfbnet

Last synced: 06 Nov 2024

https://github.com/dmmiller612/bert-extractive-summarizer

Easy to use extractive text summarization with BERT

bert coreference extractive-summarization pytorch

Last synced: 26 Dec 2024

https://github.com/patrick-kidger/torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

named-tensors python-typing pytorch shape tensors typing

Last synced: 24 Dec 2024

https://github.com/nerfstudio-project/nerfacc

A General NeRF Acceleration Toolbox in PyTorch.

instant-ngp nerf pytorch rendering

Last synced: 26 Dec 2024

https://github.com/yoshitomo-matsubara/torchdistill

A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

amazon-sagemaker-lab cifar10 cifar100 coco colab-notebook glue google-colab image-classification imagenet knowledge-distillation natural-language-processing nlp object-detection pascal-voc pytorch semantic-segmentation transformer

Last synced: 25 Dec 2024

https://github.com/ymcui/chinese-electra

Pre-trained Chinese ELECTRA(中文ELECTRA预训练模型)

bert chinese chinese-electra electra language-model nlp pre-trained-model pytorch tensorflow

Last synced: 22 Dec 2024

https://github.com/torchgan/torchgan

Research Framework for easy and efficient training of GANs based on Pytorch

computer-vision deep-learning gans generative-adversarial-networks generative-model machine-learning neural-networks python python3 pytorch

Last synced: 27 Dec 2024

https://github.com/vitae-transformer/vitpose

The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose Estimation" and [TPAMI'23] "ViTPose++: Vision Transformer for Generic Body Pose Estimation"

deep-learning distillation mae pose-estimation pytorch self-supervised-learning vision-transformer

Last synced: 26 Dec 2024

https://github.com/qfgaohao/pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

object-detection open-images pytorch ssd

Last synced: 28 Dec 2024

https://github.com/ymcui/Chinese-ELECTRA

Pre-trained Chinese ELECTRA(中文ELECTRA预训练模型)

bert chinese chinese-electra electra language-model nlp pre-trained-model pytorch tensorflow

Last synced: 07 Nov 2024

https://github.com/pytorch/functorch

functorch is JAX-like composable function transforms for PyTorch.

gradients hessians pytorch

Last synced: 24 Dec 2024

https://github.com/llvm/torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.

compiler mlir pytorch

Last synced: 24 Dec 2024

https://github.com/JiaRenChang/PSMNet

Pyramid Stereo Matching Network (CVPR2018)

psmnet pytorch stereo-matching stereo-vision

Last synced: 27 Oct 2024

https://github.com/DiffEqML/torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

control-theory deep-equilibrium-models deep-learning dynamical-systems implicit-models neural-differential-equations neural-network neural-ode numerical-methods pytorch

Last synced: 30 Oct 2024

https://github.com/lxztju/pytorch_classification

利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码

cnn densenet deployment flask image-classification knn knowledge-distillation label-smoothing pytorch random-forest resnet resnext svm

Last synced: 22 Dec 2024

https://github.com/we0091234/chinese_license_plate_detection_recognition

yolov5 车牌检测 车牌识别 中文车牌识别 检测 支持12种中文车牌 支持双层车牌

deep-learning plate-recognition pytorch yolov5

Last synced: 26 Dec 2024

https://github.com/bermanmaxim/lovaszsoftmax

Code for the Lovász-Softmax loss (CVPR 2018)

image-segmentation loss-functions neural-networks pytorch

Last synced: 28 Dec 2024

https://github.com/jeshraghian/snntorch

Deep and online learning with spiking neural networks in Python

machine-learning neural-networks neuron-models neuroscience python pytorch snn spike spiking spiking-neural-networks

Last synced: 24 Dec 2024

https://github.com/BenAAndrew/Voice-Cloning-App

A Python/Pytorch app for easily synthesising human voices

deep-learning python pytorch tacotron2 text-to-speech tts voice-cloning

Last synced: 31 Oct 2024

https://github.com/TradeMaster-NTU/TradeMaster

TradeMaster is an open-source platform for quantitative trading empowered by reinforcement learning :fire: :zap: :rainbow:

finance fintech investment-strategies jupyter-notebook machine-learning python pytorch quantitative-trading reinforcement-learning stock-market trading-platform

Last synced: 08 Nov 2024

https://github.com/flashinfer-ai/flashinfer

FlashInfer: Kernel Library for LLM Serving

cuda flash-attention gpu jit large-large-models llm-inference pytorch

Last synced: 26 Dec 2024

https://github.com/miracleyoo/pytorch-lightning-template

An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板,简单易用,稍改原来Pytorch代码,即可适配Lightning。You can translate your previous Pytorch code much easier using this template, and keep your freedom to edit all the functions as well. Big-project-friendly as well. No need to rewrite your config in hydra.

best-practices config deep-learning pytorch pytorch-lightning pytorch-lightning-template template

Last synced: 22 Dec 2024

https://github.com/godweiyang/nn-cuda-example

Several simple examples for popular neural network toolkits calling custom CUDA operators.

cpp cuda neural-network python pytorch tensorflow

Last synced: 22 Dec 2024

https://github.com/manujosephv/pytorch_tabular

A standard framework for modelling Deep Learning Models for tabular data

deep-learning hacktoberfest machine-learning pytorch pytorch-lightning tabular-data

Last synced: 24 Dec 2024

https://github.com/arthur151/romp

Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]

3d-mesh-recovery bottom-up multi-person multi-person-3d-mesh-recovery pose-estimation pytorch smpl

Last synced: 27 Dec 2024

https://github.com/mit-han-lab/torchquantum

A PyTorch-based framework for Quantum Classical Simulation, Quantum Machine Learning, Quantum Neural Networks, Parameterized Quantum Circuits with support for easy deployments on real quantum computers.

deep-learning machine-learning ml-for-systems neural-network parameterized-quantum-circuit pytorch pytorch-quantum quantum quantum-computing quantum-machine-learning quantum-neural-network quantum-simulation system

Last synced: 25 Dec 2024

https://github.com/takuseno/d3rlpy

An offline deep reinforcement learning library

deep-learning deep-reinforcement-learning offline-rl pytorch

Last synced: 24 Dec 2024

https://github.com/atong01/conditional-flow-matching

TorchCFM: a Conditional Flow Matching library

continuous-normalizing-flows optimal-transport pytorch

Last synced: 25 Dec 2024

https://github.com/Arthur151/ROMP

Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]

3d-mesh-recovery bottom-up multi-person multi-person-3d-mesh-recovery pose-estimation pytorch smpl

Last synced: 28 Oct 2024

https://github.com/cysu/open-reid

Open source person re-identification library in python

deep-learning metric-learning person-reidentification python pytorch

Last synced: 22 Dec 2024

https://github.com/alibaba/federatedscope

An easy-to-use federated learning platform

federated-learning machine-learning pytorch

Last synced: 26 Dec 2024

https://github.com/DefTruth/CUDA-Learn-Notes

🎉 Modern CUDA Learn Notes with PyTorch: fp32/tf32, fp16/bf16, fp8/int8, flash_attn, rope, sgemm, sgemv, warp/block reduce, dot, elementwise, softmax, layernorm, rmsnorm.

block-reduce cuda cuda-programming elementwise flash-attention flash-attention-2 flash-attention-3 gemm gemv layernorm pytorch rmsnorm softmax triton warp-reduce

Last synced: 27 Oct 2024

https://github.com/godweiyang/NN-CUDA-Example

Several simple examples for popular neural network toolkits calling custom CUDA operators.

cpp cuda neural-network python pytorch tensorflow

Last synced: 28 Oct 2024

https://github.com/nianticlabs/simplerecon

[ECCV 2022] SimpleRecon: 3D Reconstruction Without 3D Convolutions

computer-vision cost-volume depth depth-estimation eccv2022 multi-view-stereo mvs pytorch scannet visualization

Last synced: 22 Dec 2024

https://github.com/shaoxiongji/federated-learning

A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561

deep-learning federated-learning pytorch

Last synced: 26 Dec 2024

https://github.com/pydn/comfyui-to-python-extension

A powerful tool that translates ComfyUI workflows into executable Python code.

ai-art comfyui generative-art image-generation pytorch stable-diffusion

Last synced: 27 Dec 2024

https://github.com/pydn/ComfyUI-to-Python-Extension

A powerful tool that translates ComfyUI workflows into executable Python code.

ai-art comfyui generative-art image-generation pytorch stable-diffusion

Last synced: 19 Dec 2024

https://github.com/d-li14/involution

[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator

cvpr2021 image-classification instance-segmentation involution object-detection operator pre-trained-model pytorch semantic-segmentation

Last synced: 22 Dec 2024

https://github.com/l1aoxingyu/deep-learning-project-template

A best practice for deep learning project template architecture.

chainer deep-learning keras pytorch pytorch-tutorial

Last synced: 23 Dec 2024

https://github.com/alibaba/FederatedScope

An easy-to-use federated learning platform

federated-learning machine-learning pytorch

Last synced: 16 Nov 2024

https://github.com/sail-sg/poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

image-classification mlp pooling pytorch transformer

Last synced: 22 Dec 2024

https://github.com/stared/livelossplot

Live training loss plot in Jupyter Notebook for Keras, PyTorch and others

deep-learning jupyter-notebook keras keras-visualization pytorch

Last synced: 27 Dec 2024

https://github.com/charlesshang/dcnv2

Deformable Convolutional Networks v2 with Pytorch

cnn deformable-convolutional-networks pytorch

Last synced: 27 Dec 2024

https://github.com/ai-forever/ghost

A new one shot face swap approach for image and video domains

computer-vision deep-face-swap deep-learning deepfake face-swap faceswap ghost ghost-faceswap ghost-swap pytorch

Last synced: 26 Dec 2024

https://github.com/jrzaurin/pytorch-widedeep

A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch

deep-learning images model-hub multimodal-deep-learning python pytorch pytorch-cv pytorch-nlp pytorch-tabular-data pytorch-transformers tabular-data text

Last synced: 25 Dec 2024

https://github.com/L1aoXingyu/Deep-Learning-Project-Template

A best practice for deep learning project template architecture.

chainer deep-learning keras pytorch pytorch-tutorial

Last synced: 15 Nov 2024

https://github.com/920232796/bert_seq2seq

pytorch实现 Bert 做seq2seq任务,使用unilm方案,现在也可以做自动摘要,文本分类,情感分析,NER,词性标注等任务,支持t5模型,支持GPT2进行文章续写。

autotitle bert crf gpt2 ner pytorch roberta seq2seq t5-model text-classification unilm

Last synced: 27 Dec 2024

https://github.com/mit-han-lab/data-efficient-gans

[NeurIPS 2020] Differentiable Augmentation for Data-Efficient GAN Training

data-efficient gans generative-adversarial-network image-generation neurips-2020 pytorch tensorflow

Last synced: 27 Dec 2024

https://github.com/yinboc/liif

Learning Continuous Image Representation with Local Implicit Image Function, in CVPR 2021 (Oral)

implicit-neural-representation machine-learning pytorch super-resolution

Last synced: 23 Dec 2024

https://github.com/jayleicn/animegan

A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.

dataset generative-adversarial-network pytorch

Last synced: 22 Dec 2024

https://github.com/MCG-NJU/VideoMAE

[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training

action-recognition mae masked-autoencoder neurips-2022 pytorch self-supervised-learning transformer video-analysis video-representation-learning video-transformer video-understanding vision-transformer

Last synced: 27 Oct 2024

https://github.com/wisconsinaivision/yolact_edge

The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.

edge-devices instance-segmentation pytorch real-time realtime yolactedge

Last synced: 22 Dec 2024

https://github.com/CharlesShang/DCNv2

Deformable Convolutional Networks v2 with Pytorch

cnn deformable-convolutional-networks pytorch

Last synced: 27 Oct 2024

https://github.com/nvlabs/dg-net

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:

apex cuhk-np dg-net dukemtmc-reid image-retrieval image-search market-1501 msmt17 open-reid person-reid person-reidentification pytorch re-identification

Last synced: 22 Dec 2024

https://github.com/jayleicn/animeGAN

A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.

dataset generative-adversarial-network pytorch

Last synced: 12 Nov 2024

https://github.com/hiyouga/FastEdit

🩹Editing large language models within 10 seconds⚡

bloom chatbots chatgpt falcon gpt large-language-models llama llms pytorch transformers

Last synced: 09 Nov 2024

https://github.com/hiyouga/fastedit

🩹Editing large language models within 10 seconds⚡

bloom chatbots chatgpt falcon gpt large-language-models llama llms pytorch transformers

Last synced: 28 Dec 2024

https://github.com/kuaikuaikim/dface

Deep learning face detection and recognition, implemented by pytorch. (pytorch实现的人脸检测和人脸识别)

deeplearning facedetection facerecognition mtcnn mtcnn-pytorch pytorch

Last synced: 28 Dec 2024

https://github.com/NVlabs/DG-Net

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:

apex cuhk-np dg-net dukemtmc-reid image-retrieval image-search market-1501 msmt17 open-reid person-reid person-reidentification pytorch re-identification

Last synced: 04 Nov 2024

https://github.com/jeff-sjtu/hybrik

Official code of "HybrIK: A Hybrid Analytical-Neural Inverse Kinematics Solution for 3D Human Pose and Shape Estimation", CVPR 2021

3d-pose-estimation cvpr cvpr21 inverse-kinematics pose-estimation pytorch smpl

Last synced: 22 Dec 2024

https://github.com/gengyanlei/fire-smoke-detect-yolov4

fire-smoke-detect-yolov4-yolov5 and fire-smoke-detection-dataset 火灾检测,烟雾检测

darknet fire-detection fire-smoke-detection-dataset pytorch smoke-detection yolov4 yolov5

Last synced: 22 Dec 2024

https://github.com/SKTBrain/KoBERT

Korean BERT pre-trained cased (KoBERT)

bert korean-nlp language-model nlp pytorch transformers

Last synced: 09 Nov 2024

https://github.com/kwea123/ngp_pl

Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)

3d-reconstruction cuda instant-ngp nerf novel-view-synthesis pytorch pytorch-lightning

Last synced: 22 Dec 2024

https://github.com/WisconsinAIVision/yolact_edge

The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.

edge-devices instance-segmentation pytorch real-time realtime yolactedge

Last synced: 28 Oct 2024

https://github.com/zalandoresearch/pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend

deepar lstnet n-beats probabilistic pytorch time-series

Last synced: 22 Dec 2024

https://github.com/kengz/slm-lab

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".

a2c a3c benchmark deep-reinforcement-learning dqn policy-gradient ppo pytorch reinforcement-learning sac

Last synced: 22 Dec 2024