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/DeepGraphLearning/torchdrug

A powerful and flexible machine learning platform for drug discovery

deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 02 Aug 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 Jul 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: 31 Jul 2024

https://github.com/breezedeus/Pix2Text

An Open-Source Python3 tool for recognizing layouts, tables, math formulas (LaTeX), and text in images, converting them into Markdown format. A free alternative to Mathpix, empowering seamless conversion of visual content into text-based representations. 80+ languages are supported.

image-to-markdown latex latex-pdf layout-analysis math-formula math-formula-recognition math-ocr mathpix ocr python pytorch table-ocr

Last synced: 02 Aug 2024

https://github.com/filipradenovic/cnnimageretrieval-pytorch

CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch

cnn convolutional-neural-networks image-retrieval python pytorch

Last synced: 31 Jul 2024

https://github.com/cszn/DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)

image-denoising jpeg-deblocking keras-tensorflow matconvnet pytorch residual-learning super-resolution

Last synced: 01 Aug 2024

https://github.com/GOATmessi7/RFBNet

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

detection mobilenet pytorch rfbnet

Last synced: 01 Aug 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: 01 Aug 2024

https://github.com/vturrisi/solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

barlow-twins byol contrastive-learning deepcluster dino mae masked-input-prediction moco nnclr nvidia-dali pytorch pytorch-lightning ressl self-supervised-learning simclr simsiam swav transformer-models vibcreg vicreg

Last synced: 01 Aug 2024

https://github.com/JiaRenChang/PSMNet

Pyramid Stereo Matching Network (CVPR2018)

psmnet pytorch stereo-matching stereo-vision

Last synced: 31 Jul 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: 01 Aug 2024

https://github.com/nod-ai/SHARK

SHARK - High Performance Machine Learning Distribution

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

Last synced: 01 Aug 2024

https://github.com/pytorch/functorch

functorch is JAX-like composable function transforms for PyTorch.

gradients hessians pytorch

Last synced: 02 Aug 2024

https://github.com/n00mkrad/flowframes

Flowframes Windows GUI for video interpolation using DAIN (NCNN) or RIFE (CUDA/NCNN)

dain gui ncnn pytorch rife video-interpolation

Last synced: 31 Jul 2024

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

Easy to use extractive text summarization with BERT

bert coreference extractive-summarization pytorch

Last synced: 03 Aug 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 Jul 2024

https://github.com/UKPLab/beir

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

ance benchmark bert colbert dataset deep-learning dpr elasticsearch information-retrieval nlp passage-retrieval pytorch question-generation retrieval retrieval-models sbert sentence-transformers use-qa zero-shot-retrieval

Last synced: 04 Aug 2024

https://github.com/beir-cellar/beir

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

ance benchmark bert colbert dataset deep-learning dpr elasticsearch information-retrieval nlp passage-retrieval pytorch question-generation retrieval retrieval-models sbert sentence-transformers use-qa zero-shot-retrieval

Last synced: 01 Aug 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: 31 Jul 2024

https://github.com/deepmodeling/deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

ase c computational-chemistry cpp cuda deep-learning deepmd ipi lammps materials-science molecular-dynamics nodejs potential-energy python pytorch rocm tensorflow

Last synced: 31 Jul 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: 02 Aug 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: 01 Aug 2024

https://github.com/modelscope/data-juicer

A one-stop data processing system to make data higher-quality, juicier, and more digestible for LLMs! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷为大语言模型提供更高质量、更丰富、更易”消化“的数据!

chinese data-analysis data-science data-visualization dataset gpt gpt-4 instruction-tuning large-language-models llama llava llm llms multi-modal nlp opendata pre-training pytorch sora streamlit

Last synced: 01 Aug 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: 31 Jul 2024

https://github.com/intel/intel-extension-for-pytorch

A Python package for extending the official PyTorch that can easily obtain performance on Intel platform

deep-learning intel machine-learning neural-network pytorch quantization

Last synced: 01 Aug 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: 31 Jul 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: 31 Jul 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: 31 Jul 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: 02 Aug 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: 03 Aug 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: 31 Jul 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: 01 Aug 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: 31 Jul 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: 01 Aug 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: 31 Jul 2024

https://github.com/SKTBrain/KoBERT

Korean BERT pre-trained cased (KoBERT)

bert korean-nlp language-model nlp pytorch transformers

Last synced: 02 Aug 2024

https://github.com/takuseno/d3rlpy

An offline deep reinforcement learning library

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

Last synced: 02 Aug 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: 03 Aug 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: 01 Aug 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: 02 Aug 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: 01 Aug 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: 03 Aug 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: 01 Aug 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: 03 Aug 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: 31 Jul 2024

https://github.com/alibaba/FederatedScope

An easy-to-use federated learning platform

federated-learning machine-learning pytorch

Last synced: 03 Aug 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: 01 Aug 2024

https://github.com/ShannonAI/service-streamer

Boosting your Web Services of Deep Learning Applications.

bert deep-learning model-deployment pytorch tensorflow web

Last synced: 01 Aug 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: 31 Jul 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: 31 Jul 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: 31 Jul 2024

https://github.com/AshwinRJ/Federated-Learning-PyTorch

Implementation of Communication-Efficient Learning of Deep Networks from Decentralized Data

deep-learning distributed-computing federated-learning python pytorch

Last synced: 02 Aug 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: 02 Aug 2024

https://github.com/microsoft/mup

maximal update parametrization (µP)

deep-learning machine-learning mup mutransfer python pytorch transformers

Last synced: 01 Aug 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: 02 Aug 2024

https://github.com/huggingface/hmtl

🌊HMTL: Hierarchical Multi-Task Learning - A State-of-the-Art neural network model for several NLP tasks based on PyTorch and AllenNLP

multi-task-learning natural-language-processing nlp pytorch

Last synced: 01 Aug 2024

https://github.com/oscarknagg/few-shot

Repository for few-shot learning machine learning projects

few-shot-learning machine-learning maml meta-learning miniimagenet omniglot pytorch research

Last synced: 01 Aug 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: 03 Aug 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: 03 Aug 2024

https://github.com/abhishekkrthakur/tez

Tez is a super-simple and lightweight Trainer for PyTorch. It also comes with many utils that you can use to tackle over 90% of deep learning projects in PyTorch.

deep-learning deep-neural-networks neural-networks pytorch tez

Last synced: 01 Aug 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: 31 Jul 2024

https://github.com/DengBoCong/nlp-paper

自然语言处理领域下的相关论文(附阅读笔记),复现模型以及数据处理等(代码含TensorFlow和PyTorch两版本)

bert dialogue nlp nlp-machine-learning paper pytorch speech tensorflow2

Last synced: 03 Aug 2024

https://github.com/PyRetri/PyRetri

Open source deep learning based unsupervised image retrieval toolbox built on PyTorch🔥

deep-learning image-retrieval pytorch toolbox

Last synced: 31 Jul 2024

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

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

acceleration pytorch

Last synced: 31 Jul 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: 07 Aug 2024

https://github.com/Pointcept/Pointcept

Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT (CVPR'24), OA-CNNs (CVPR'24), MSC (CVPR'23)

3d-vision point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/Xilinx/brevitas

Brevitas: neural network quantization in PyTorch

brevitas deep-learning fpga hardware-acceleration neural-networks ptq pytorch qat quantization xilinx

Last synced: 02 Aug 2024

https://github.com/lightaime/deep_gcns_torch

Pytorch Repo for DeepGCNs (ICCV'2019 Oral, TPAMI'2021), DeeperGCN (arXiv'2020) and GNN1000(ICML'2021): https://www.deepgcns.org

3d-point-clouds bioinformatics cheminformatics computer-vision data-mining deep-gcns deep-learning geometric-deep-learning graph-convolutional-networks graph-neural-networks pytorch science-research social-network

Last synced: 31 Jul 2024

https://github.com/lartpang/PyTorchTricks

Some tricks of pytorch... :star:

pytorch pytorch-trick pytorch-tutorial tricks

Last synced: 31 Jul 2024

https://github.com/IAHispano/Applio

VITS-based Voice Conversion focused on simplicity, quality and performance.

ai applio pytorch rvc speech speech-to-speech text-to-speech vc vits voice voice-clone voice-cloning voice-conversion

Last synced: 03 Aug 2024

https://github.com/InterDigitalInc/CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

compression deep-learning deep-neural-networks machine-learning neural-network python pytorch

Last synced: 02 Aug 2024

https://github.com/dsgiitr/graph_nets

PyTorch Implementation and Explanation of Graph Representation Learning papers: DeepWalk, GCN, GraphSAGE, ChebNet & GAT.

chebyshev-polynomials deepwalk graph-attention-networks graph-convolutional-networks graph-embedding graph-representation-learning graph-sage node-embedding pytorch

Last synced: 02 Aug 2024