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: 14 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/JiaRenChang/PSMNet

Pyramid Stereo Matching Network (CVPR2018)

psmnet pytorch stereo-matching stereo-vision

Last synced: 27 Oct 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: 14 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: 15 Oct 2024

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

A General NeRF Acceleration Toolbox in PyTorch.

instant-ngp nerf pytorch rendering

Last synced: 14 Oct 2024

https://github.com/pytorch/functorch

functorch is JAX-like composable function transforms for PyTorch.

gradients hessians pytorch

Last synced: 07 Oct 2024

https://github.com/bermanmaxim/lovaszsoftmax

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

image-segmentation loss-functions neural-networks pytorch

Last synced: 14 Oct 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/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: 15 Oct 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/flashinfer-ai/flashinfer

FlashInfer: Kernel Library for LLM Serving

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

Last synced: 31 Oct 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: 15 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: 10 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: 29 Oct 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: 15 Oct 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/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: 15 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: 29 Oct 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: 14 Oct 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: 29 Oct 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/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: 15 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/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: 15 Oct 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: 29 Oct 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: 30 Oct 2024

https://github.com/takuseno/d3rlpy

An offline deep reinforcement learning library

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

Last synced: 22 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: 29 Oct 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: 15 Oct 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: 18 Oct 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: 14 Oct 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: 01 Nov 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/CharlesShang/DCNv2

Deformable Convolutional Networks v2 with Pytorch

cnn deformable-convolutional-networks pytorch

Last synced: 27 Oct 2024

https://github.com/we0091234/chinese_license_plate_detection_recognition

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

deep-learning plate-recognition pytorch yolov5

Last synced: 15 Oct 2024

https://github.com/charlesshang/dcnv2

Deformable Convolutional Networks v2 with Pytorch

cnn deformable-convolutional-networks pytorch

Last synced: 14 Oct 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: 14 Oct 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: 15 Oct 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/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: 10 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/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: 29 Oct 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: 29 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: 04 Nov 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: 29 Oct 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: 15 Oct 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: 10 Oct 2024

https://github.com/alibaba/federatedscope

An easy-to-use federated learning platform

federated-learning machine-learning pytorch

Last synced: 09 Oct 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/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/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: 29 Oct 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: 02 Nov 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: 09 Oct 2024

https://github.com/open-compass/vlmevalkit

Open-source evaluation toolkit of large vision-language models (LVLMs), support ~100 VLMs, 40+ benchmarks

chatgpt claude clip computer-vision evaluation gemini gpt gpt-4v gpt4 large-language-models llava llm multi-modal openai openai-api pytorch qwen vit vqa

Last synced: 08 Nov 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: 14 Oct 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: 29 Oct 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: 15 Oct 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/zalandoresearch/pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend

deepar lstnet n-beats probabilistic pytorch time-series

Last synced: 10 Nov 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/ShannonAI/service-streamer

Boosting your Web Services of Deep Learning Applications.

bert deep-learning model-deployment pytorch tensorflow web

Last synced: 04 Nov 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: 15 Oct 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: 07 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: 14 Oct 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: 14 Oct 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: 30 Oct 2024

https://github.com/charliedream1/ai_quant_trade

股票AI操盘手:从学习、模拟到实盘,一站式平台。包含股票知识、策略实例、因子挖掘、传统策略、机器学习、深度学习、强化学习、图网络、高频交易、C++部署和聚宽实例代码等,可以方便学习、模拟及实盘交易

cpp jupyter-notebook keras mlflow python pytorch sklearn tensorflow trading-bot trading-platform trading-strategies

Last synced: 13 Oct 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: 30 Oct 2024

https://github.com/jhj0517/Whisper-WebUI

A Web UI for easy subtitle using whisper model.

ai gradio open-source python pytorch web-ui whisper

Last synced: 20 Oct 2024