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/wangleiphy/tensorgrad

Differentiable Programming Tensor Networks

automatic-differentiation pytorch tensor-networks

Last synced: 05 Aug 2024

https://github.com/assassint2017/abdominal-multi-organ-segmentation

abdominal multi-organ segmentation using pytorch

multi-organ-segmentation pytorch segmentation

Last synced: 05 Aug 2024

https://github.com/IntelLabs/matsciml

Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery supporting widely used materials science datasets, and built on top of PyTorch Lightning, the Deep Graph Library, and PyTorch Geometric.

ai dgl pytorch pytorch-lightning

Last synced: 04 Aug 2024

https://github.com/warner-benjamin/commented-transformers

Highly commented implementations of Transformers in PyTorch

bert gpt pytorch transformers

Last synced: 28 Oct 2024

https://github.com/SherylHYX/pytorch_geometric_signed_directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

deep-learning directed-networks gnn graph-neural-netowrks machine-learning networks python pytorch pytorch-geometric signed-networks

Last synced: 17 Nov 2024

https://github.com/EthanRosenthal/torchmf

matrix factorization in PyTorch

matrix-factorization pytorch recommender-systems

Last synced: 11 Nov 2024

https://github.com/dandelin/dynamic-memory-networks-plus-pytorch

Implementation of Dynamic memory networks plus in Pytorch

dmn language-model python pytorch question-answering

Last synced: 06 Nov 2024

https://github.com/scaleoutsystems/fedn

FEDn: An open framework and a managed platform for real-world federated machine learning.

edge-ai edge-ml federated-learning federated-machine-learning fedml fleet-learning keras-tensorflow pytorch scikit-learn tensorflow

Last synced: 29 Oct 2024

https://github.com/microsoft/adamix

This is the implementation of the paper AdaMix: Mixture-of-Adaptations for Parameter-efficient Model Tuning (https://arxiv.org/abs/2205.12410).

adapter bert dart glue gpt-2 nlg nlp nlu parameter-efficient pytorch roberta webnlg

Last synced: 07 Oct 2024

https://github.com/BiomedSciAI/fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)

ai cmmd collaboration ct deep-learning fuse fuse-med-ml fusemedml hacktoberfest healthcare isic knight-challenge machine-learning medical medical-imaging multimodality python pytorch stoic vision

Last synced: 15 Nov 2024

https://github.com/EDAPINENUT/CLCRN

This is an official Pytorch implementation of Conditional Local Convolution for Spatio-temporal Meteorological Forecasting, AAAI 2022

aaai2022 continuous-convolution graph-neural-networks meteorological-forecasting pytorch spatio-temporal spherical-geometry

Last synced: 08 Aug 2024

https://github.com/FanDady/Helmet-Detection-YoloV5

Safety helmet wearing detection on construction site based on YoloV5s-V5.0 including helmet dataset(基于YoloV5-V5.0的工地安全帽检测并且包含开源的安全帽数据集)

python pytorch yolov5s

Last synced: 05 Oct 2024

https://github.com/yurizzzzz/Helmet-Detection-YoloV5

Safety helmet wearing detection on construction site based on YoloV5s-V5.0 including helmet dataset(基于YoloV5-V5.0的工地安全帽检测并且包含开源的安全帽数据集)

python pytorch yolov5s

Last synced: 09 Nov 2024

https://github.com/jackaduma/chatglm-lora-rlhf-pytorch

A full pipeline to finetune ChatGLM LLM with LoRA and RLHF on consumer hardware. Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the ChatGLM architecture. Basically ChatGPT but with ChatGLM

chatglm chatglm-6b chatgpt deepspeed finetune gpt llama llm lora peft ppo pytorch reward-models rlhf

Last synced: 11 Nov 2024

https://github.com/borchero/pycave

Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.

gaussian-mixture-models kmeans machine-learning markov-model pytorch pytorch-lightning

Last synced: 15 Nov 2024

https://github.com/donydchen/sem2nerf

👩🏼‍🦰😺[ECCV'22] Official PyTorch Implementation of Sem2NeRF: Converting Single-View Semantic Masks to NeRFs

eccv-2022 eccv2022 gan image-generation image-manipulation image-to-image image-translation nerf pytorch semantic-to-nerf

Last synced: 16 Nov 2024

https://github.com/hosseinshn/basic-multi-task-learning

This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow

multitask-learning pytorch tensorflow

Last synced: 08 Nov 2024

https://github.com/lupantech/InterGPS

Data and Code for ACL 2021 Paper "Inter-GPS: Interpretable Geometry Problem Solving with Formal Language and Symbolic Reasoning"

mathqa pytorch

Last synced: 16 Nov 2024

https://github.com/guanhuaw/mirtorch

A PyTorch-powered differentiable image reconstruction/optimization toolbox

magnetic-resonance-imaging medical-imaging model-based pytorch

Last synced: 16 Nov 2024

https://github.com/leehanchung/cs224n

Stanford CS224n: Natural Language Processing with Deep Learning, Winter 2020

2020 cs224n cs224n-assignment-solutions cs224nwinter2020 natural-language-processing pytorch stanford stanford-nlp

Last synced: 10 Nov 2024

https://github.com/rsnk96/Ubuntu-Setup-Scripts

Scripts to help you set up your Ubuntu quickly, especially if you're in any subfield of Data Science or AI!

anaconda cuda deep-learning deeplearning dl ffmpeg installers ml opencv python pytorch tensorflow tensorflow-setup ubuntu zsh

Last synced: 06 Nov 2024

https://github.com/kefirski/pytorch_NEG_loss

NEG loss implemented in pytorch

python pytorch word2vec

Last synced: 14 Nov 2024

https://github.com/sherylhyx/pytorch_geometric_signed_directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

deep-learning directed-networks gnn graph-neural-netowrks machine-learning networks python pytorch pytorch-geometric signed-networks

Last synced: 12 Nov 2024

https://github.com/jingtaozhan/DRhard

SIGIR'21: Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track.

information-retrieval pytorch web-search

Last synced: 16 Nov 2024

https://github.com/rasbt/faster-pytorch-blog

Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracy

deeplearning lightning pytorch transformers

Last synced: 01 Nov 2024

https://github.com/szagoruyko/binary-wide-resnet

PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)

pytorch wide-residual-networks

Last synced: 29 Oct 2024

https://github.com/andrewekhalel/edafa

Test Time Augmentation (TTA) wrapper for computer vision tasks: segmentation, classification, super-resolution, ... etc.

augmentation classification computer-vision keras pytorch segmentation super-resolution tensorflow wrapper

Last synced: 15 Nov 2024

https://github.com/deepcam-cn/face-anti-spoofing.pytorch

Train code of face anti-spoofing with a single RGB frame

face-anti-spoofing pytorch

Last synced: 07 Nov 2024

https://github.com/dahyun-kang/ifsl

[CVPR'22] Official PyTorch implementation of Integrative Few-Shot Learning for Classification and Segmentation

computer-vision cvpr2022 deep-learning few-shot-classification few-shot-learning few-shot-segmentation pytorch pytorch-lightning

Last synced: 15 Nov 2024

https://github.com/emilemathieu/pvae

code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".

hierarchical-data hyperbolic-geometry poincare-embeddings pytorch vae

Last synced: 13 Oct 2024

https://github.com/astorfi/3d-convolutional-speaker-recognition-pytorch

:speaker: Deep Learning & 3D Convolutional Neural Networks for Speaker Verification

convolutional-neural-networks python pytorch speaker-verification

Last synced: 01 Nov 2024

https://github.com/hkchengrex/mask-propagation

[CVPR 2021] MiVOS - Mask Propagation module. Reproduced STM (and better) with training code :star2:. Semi-supervised video object segmentation evaluation.

computer-vision cvpr2021 deep-learning pytorch segmentation video-object-segmentation video-segmentation

Last synced: 10 Nov 2024

https://github.com/nmichlo/disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib

autoencoders configurable datasets disentangled-representations disentanglement metric-learning metrics python python3 pytorch pytorch-lightning representation-learning vae

Last synced: 15 Nov 2024

https://github.com/astorfi/sequence-to-sequence-from-scratch

:speech_balloon: Sequence to Sequence from Scratch Using Pytorch

deep-learning machine-learning python pytorch

Last synced: 01 Nov 2024

https://github.com/zhiqwang/sightseq

Computer vision tools for fairseq, containing PyTorch implementation of text recognition and object detection

attention crnn ctc densenet faster-rcnn image-captioning mobilenet object-detection ocr pytorch scene-texts text-recognition transformer

Last synced: 07 Aug 2024

https://github.com/gaohongkui/tplinker-ner

TPlinker for NER 中文/英文命名实体识别

chinese-ner ner pytorch tplinker tplinker-ner

Last synced: 12 Nov 2024

https://github.com/sayakpaul/robustness-vit

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

computer-vision jax pytorch robustness self-attention tensorflow transformers

Last synced: 01 Nov 2024

https://github.com/rdspring1/pytorch_gbw_lm

PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset

deep-learning gpu language-model lstm machine-learning nlp pytorch torch torch-gbw

Last synced: 29 Oct 2024

https://github.com/gaohongkui/TPLinker-NER

TPlinker for NER 中文/英文命名实体识别

chinese-ner ner pytorch tplinker tplinker-ner

Last synced: 06 Nov 2024

https://github.com/yeyupiaoling/pytorch-mobilefacenet

Pytorch实现的人脸识别明细MobileFaceNet模型,在预测使用MTCNN检测人脸,然后使用MobileFaceNet模型识别。

face-recognition mobilefacenet mtcnn pytorch

Last synced: 01 Nov 2024

https://github.com/developmentseed/fastai-serving

A Docker image for serving fast.ai models, mimicking the API of Tensorflow Serving

deep-learning docker machine-learning python pytorch

Last synced: 05 Aug 2024

https://github.com/leverxgroup/esrgan

Enhanced SRGAN. Champion PIRM Challenge on Perceptual Super-Resolution

computer-vision deep-neural-networks generative-adversarial-network image-processing pytorch super-resolution

Last synced: 26 Oct 2024

https://github.com/Jasonkks/PTTR

Pytorch Implementation of PTTR: Relational 3D Point Cloud Object Tracking with Transformer

autonomous-driving point-cloud pytorch tracking

Last synced: 28 Oct 2024

https://github.com/lartpang/ZoomNet

Zoom In and Out: A Mixed-scale Triplet Network for Camouflaged Object Detection, CVPR 2022

camouflaged-object-detection cod codeforpaper cvpr cvpr2022 paper papercode python pytorch rgbcod rgbsod saliency-detection salient-object-detection sod

Last synced: 15 Nov 2024

https://github.com/yuweihao/KERN

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

computer-vision graph-neural-network pytorch scene-graph scene-graph-generation

Last synced: 02 Nov 2024

https://github.com/JusperLee/Deep-Clustering-for-Speech-Separation

Pytorch implements Deep Clustering: Discriminative Embeddings For Segmentation And Separation

deep-clustering pytorch segmentation speech-separation

Last synced: 02 Nov 2024

https://github.com/cceyda/torchserve-dashboard

Management Dashboard for Torchserve

ml-deployment pytorch streamlit torchserve

Last synced: 31 Oct 2024

https://github.com/marcoslucianops/deepstream-yolo-pose

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 application for YOLO-Pose models

deepstream nvidia nvidia-deepstream-sdk object-detection pose-estimation pytorch tensorrt ultralytics yolo yolov8

Last synced: 16 Nov 2024

https://github.com/yuweihao/kern

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

computer-vision graph-neural-network pytorch scene-graph scene-graph-generation

Last synced: 06 Nov 2024

https://github.com/amazon-science/tanl

Structured Prediction as Translation between Augmented Natural Languages

deep-learning natural-language-processing pytorch

Last synced: 12 Nov 2024

https://github.com/xuyxu/deep-clustering-network

PyTorch Implementation of "Towards K-Means-Friendly Spaces: Simultaneous Deep Learning and Clustering," Bo Yang et al., ICML'2017.

clustering deep-learning pytorch

Last synced: 08 Nov 2024

https://github.com/qq44642754a/Yolov5_ros

Real-time object detection with ROS, based on YOLOv5 and PyTorch (基于 YOLOv5的ROS实时对象检测)

pytorch ros yolov5

Last synced: 09 Nov 2024

https://github.com/lonepatient/bilstm-crf-ner-pytorch

This repo contains a PyTorch implementation of a BiLSTM-CRF model for named entity recognition task.

bilstm-crf crf lstm ner nlp pytorch

Last synced: 06 Nov 2024

https://github.com/nvidia/clara-train-examples

Example notebooks demonstrating how to use Clara Train to build Medical Imaging Deep Learning models

automl clara-train deep-learning healthcare-imaging medical-imaging-computing medical-imaging-processing python3 pytorch tcia-dac

Last synced: 29 Oct 2024

https://github.com/graykode/toeicbert

TOEIC(Test of English for International Communication) solving using pytorch-pretrained-BERT model.

ai bert deep-learning lm mask nlp pytorch pytorch-pretrained toeic

Last synced: 01 Nov 2024

https://github.com/kedreamix/pytorch-image-classification

用于pytorch的图像分类,包含多种模型方法,比如AlexNet,VGG,GoogleNet,ResNet,DenseNet等等,包含可完整运行的代码。除此之外,也有colab的在线运行代码,可以直接在colab在线运行查看结果。也可以迁移到自己的数据集进行迁移学习。

alexnet cifar10 densenet goolenet inception mnasnet mobilenet pytorch resnet shufflenet vgg

Last synced: 17 Nov 2024

https://github.com/vra/flopth

A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.

cnn deep-learning deeplearning flop flops pip pypi python python3 pytorch visualization

Last synced: 14 Nov 2024

https://github.com/sniklaus/pytorch-extension

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors

cuda cupy deep-learning python pytorch

Last synced: 14 Nov 2024

https://github.com/lupantech/intergps

Data and Code for ACL 2021 Paper "Inter-GPS: Interpretable Geometry Problem Solving with Formal Language and Symbolic Reasoning"

mathqa pytorch

Last synced: 05 Nov 2024

https://github.com/ise-uiuc/nnsmith

Automatic DNN generation for fuzzing and more

compiler deep-learning fuzzing machine-learning pytorch tensorflow

Last synced: 11 Oct 2024

https://github.com/chrhenning/hypnettorch

Package for working with hypernetworks in PyTorch.

deep-learning hypernet hypernets hypernetwork hypernetworks neural-network python pytorch torch

Last synced: 11 Nov 2024

https://github.com/um-arm-lab/pytorch_volumetric

Volumetric structures such as voxels and SDFs implemented in pytorch

chamfer-distance kinematics pytorch robot sdf signed-distance-field signed-distance-functions voxel voxel-grid voxels

Last synced: 16 Nov 2024

https://github.com/lucidrains/pytorch-custom-utils

Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new AI research

accelerate pytorch

Last synced: 12 Nov 2024

https://github.com/AlbertSuarez/object-cut

✂️ Cut the main object of an image automagically

basnet connexion crop-image image-processing openapi pytorch

Last synced: 26 Oct 2024

https://github.com/bradyfu/dvg

[NeurIPS 2019] Dual Variational Generation for Low Shot Heterogeneous Face Recognition

face-recognition heterogeneous-face-recognition pytorch

Last synced: 16 Nov 2024

https://github.com/alibaba/graphlearn-for-pytorch

A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.

deep-learning distributed gpu graph-neural-networks pytorch

Last synced: 12 Nov 2024

https://github.com/bharathgs/NALU

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units paper by trask et.al

accumulator alu arithmetic deep-learning deepmind logic machine-learning nalu neural-network pytorch

Last synced: 14 Nov 2024

https://github.com/ajayarunachalam/msda

Library for multi-dimensional, multi-sensor, uni/multivariate time series data analysis, unsupervised feature selection, unsupervised deep anomaly detection, and prototype of explainable AI for anomaly detector

anamoly-detection-using-graphs anomaly-detection correlation data-analysis deep-learning deep-neural-networks explainable-artificial-intelligence feature-engineering feature-selection multidimensional-data multisensor python pytorch sensor sensor-data signal-processing tabular-data time-series variation visualization

Last synced: 06 Nov 2024

https://github.com/bharathgs/nalu

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units paper by trask et.al

accumulator alu arithmetic deep-learning deepmind logic machine-learning nalu neural-network pytorch

Last synced: 15 Nov 2024