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/bloomberg/minilmv2.bb

Our open source implementation of MiniLMv2 (https://aclanthology.org/2021.findings-acl.188)

distillation language-model model-compression model-distillation python pytorch transformers

Last synced: 09 Nov 2024

https://github.com/speediedan/finetuning-scheduler

A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.

artificial-intelligence fine-tuning finetuning machine-learning neural-networks pytorch pytorch-lightning superglue transfer-learning

Last synced: 09 Nov 2024

https://github.com/guocheng18/transformer-encoder

Implementation of Transformer encoder in PyTorch

easy-to-use encoder pytorch transformer

Last synced: 16 Nov 2024

https://github.com/ppogg/retinaface_ghost

This is a project based on retinaface face detection, including ghostnet and mobilenetv3

ghostnet mobilenetv3 pytorch retinaface

Last synced: 20 Nov 2024

https://github.com/ajithvcoder/caae-age_progression_regression_pytorch

Code for Age progression/Regression, http://web.eecs.utk.edu/~zzhang61/docs/papers/2017_CVPR_Age.pdf

aging gan pytorch utkface

Last synced: 13 Nov 2024

https://github.com/hustvl/gneuvox

GNeuVox: Generalizable Neural Voxels for Fast Human Radiance Fields

3d computer-graphics computer-vision humannerf monocular nerf pytorch view-synthesis

Last synced: 09 Nov 2024

https://github.com/liyuanlucasliu/torch-scope

A Toolkit for Training, Tracking, Saving Models and Syncing Results

gpu-memory logger pytorch tensorboard toolkit

Last synced: 09 Nov 2024

https://github.com/mboudiaf/pytorch-meta-dataset

A non-official 100% PyTorch implementation of META-DATASET benchmark for few-shot classification

few-shot-classifcation few-shot-learning meta-dataset pytorch tfrecorddataset

Last synced: 15 Nov 2024

https://github.com/snakers4/playing_with_vae

Comparing FC VAE / FCN VAE / PCA / UMAP on MNIST / FMNIST

beginner embedding fashion-mnist mnist pca python3 pytorch tutorial umap variational-autoencoder

Last synced: 02 Nov 2024

https://github.com/juanmc2005/speakerembeddinglosscomparison

Companion repository for the paper "A Comparison of Metric Learning Loss Functions for End-to-End Speaker Verification" published at SLSP 2020

additive-angular-margin-loss end-to-end-machine-learning metric-learning pytorch sincnet speaker-embedding speaker-verification x-vector

Last synced: 28 Oct 2024

https://github.com/fregu856/ebms_3dod

Official implementation of "Accurate 3D Object Detection using Energy-Based Models", CVPR Workshops 2021.

3d-object-detection computer-vision deep-learning energy-based-model machine-learning object-detection pytorch

Last synced: 28 Oct 2024

https://github.com/sea-snell/grokking

unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"

artificial-intelligence deep-learning grokking neural-network python pytorch transformer transformer-models

Last synced: 27 Oct 2024

https://github.com/MrtnMndt/OpenVAE_ContinualLearning

Open-source code for our paper: Unified Probabilistic Deep Continual Learning through Generative Replay and Open Set Recognition

continual-learning deep-learning open-set-recognition pytorch variational-autoencoder

Last synced: 15 Nov 2024

https://github.com/nishiwen1214/dropattack

Code of paper: "DropAttack: A Masked Weight Adversarial Training Method to Improve Generalization of Neural Networks"

adversarial-learning dropattack generalization pytorch pytorch-implementation text-classification

Last synced: 19 Nov 2024

https://github.com/maum-ai/pnlp-mixer

Unofficial PyTorch Implementation for pNLP-Mixer: an Efficient all-MLP Architecture for Language (https://arxiv.org/abs/2202.04350)

mlp-mixer nlp pytorch pytorch-lightning

Last synced: 11 Nov 2024

https://github.com/swyoon/pytorch-minimal-gaussian-process

A minimal implementation of Gaussian process regression in PyTorch

gaussian-process gaussian-process-regression gaussian-processes pytorch

Last synced: 11 Nov 2024

https://github.com/nikhilbarhate99/deterministic-gail-pytorch

PyTorch implementation of Deterministic Generative Adversarial Imitation Learning (GAIL) for Off Policy learning

deep-reinforcement-learning gail generative-adversarial-network gym imitation-learning openai-gym pytorch pytorch-gan pytorch-implementation reinforcement-learning

Last synced: 13 Nov 2024

https://github.com/1adrianb/unsupervised-face-representation

Implementation for Pre-training strategies and datasets for facial representation learning, ECCV 2022

emotion-recognition face-alignment face-recognition face-representation-learning pytorch representation-learning unsupervised-learning

Last synced: 14 Nov 2024

https://github.com/kevinzakka/vae-pytorch

AE and VAE Playground in PyTorch

autoencoder pytorch variational-autoencoder

Last synced: 28 Oct 2024

https://github.com/jpuigcerver/pytorch-baidu-ctc

PyTorch bindinga for Baidu's Warp-CTC

ctc-loss cuda pytorch

Last synced: 01 Oct 2024

https://github.com/soumik12345/zero-dce

Pytorch implementation of Zero-Reference Deep Curve Estimation for Low-Light Image Enhancement

deep-learning hdr low-light-enhance low-light-image-enhancement pytorch zero-shot-learning

Last synced: 02 Nov 2024

https://github.com/cxhernandez/cellcount

A Convolutional Neural Network for Segmenting and Counting Cells in Microscopy Images

biology computer-vision python pytorch segmentation

Last synced: 28 Oct 2024

https://github.com/Lightning-Universe/lightning-ColossalAI

Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI

lightning python pytorch

Last synced: 09 Nov 2024

https://github.com/Lee-Gihun/MixCo-Mixup-Contrast

(NeurIPS 2020 Workshop on SSL) Official Implementation of "MixCo: Mix-up Contrastive Learning for Visual Representation"

deep-learning mix-up pytorch self-supervised-learning

Last synced: 13 Nov 2024

https://github.com/rmeli/gnina-torch

🔥 PyTorch implementation of GNINA scoring function for molecular docking

deep-learning docking drug-discovery gnina pytorch pytorch-ignite scoring-functions

Last synced: 15 Nov 2024

https://github.com/godka/pensieve-ppo

The simplest implementation of Pensieve (SIGCOMM' 17) via state-of-the-art RL algorithms, including PPO, DQN, SAC, and support for both TensorFlow and PyTorch.

a2c deep-learning dqn pensieve ppo pytorch reinforcement-learning tensorflow

Last synced: 19 Nov 2024

https://github.com/KushajveerSingh/resize_network_cv

PyTorch implementation of the paper "Learning to Resize Images for Computer Vision Tasks" on Imagenette and Imagewoof datasets

computer-vision pytorch pytorch-implementation pytorch-lightning

Last synced: 15 Nov 2024

https://github.com/Lightning-AI/lightning-colossalai

Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI

lightning python pytorch

Last synced: 05 Oct 2024

https://github.com/lonepatient/bert-attribute-value-extract

A Pytorch implementation of "Scaling Up Open Tagging from Tens to Thousands: Comprehension Empowered Attribute Value Extraction from Product Title" (ACL 2019).

attention attribute-value-extract bert crf lstm pytorch

Last synced: 06 Nov 2024

https://github.com/searchableai/kitanaqa

KitanaQA: Adversarial training and data augmentation for neural question-answering models

adversarial-attacks adversarial-training bert data-augmentation ml-automation natural-language-processing nlp pytorch question-answering transformer

Last synced: 13 Oct 2024

https://github.com/amanchadha/frvsr-gan

A Novel Approach to Video Super-Resolution using Frame Recurrence and Generative Adversarial Networks | Python3 | PyTorch | OpenCV2 | GANs | CNNs

adjacent-frames cnn cnns convolutional-neural-networks frame-recurrence frvsr-gan gan gans generative-adversarial-network opencv psnr python python3 pytorch recurrent-neural-network sisr srgan ssim super-resolution video-super-resolution

Last synced: 05 Nov 2024

https://github.com/saturncloud/dask-pytorch-ddp

dask-pytorch-ddp is a Python package that makes it easy to train PyTorch models on dask clusters using distributed data parallel.

computer-vision dask deep-learning distributed-computing machine-learning nlp pytorch

Last synced: 15 Nov 2024

https://github.com/seungeunrho/football-paris

The exact codes used by the team "liveinparis" at the kaggle football competition ranked 6th/1141

gfootball kaggle liveinparis ppo pytorch reinforcement-learning self-play

Last synced: 09 Nov 2024

https://github.com/gokumohandas/oreilly-pytorch

🔥 Introductory PyTorch tutorials with OReilly Media.

machine-learning pytorch

Last synced: 09 Nov 2024

https://github.com/cdancette/rubi.bootstrap.pytorch

NeurIPS 2019 Paper: RUBi : Reducing Unimodal Biases for Visual Question Answering

bias bias-reduction deep-learning pytorch vqa

Last synced: 07 Nov 2024

https://github.com/pmixer/tisasrec.pytorch

https://github.com/JiachengLi1995/TiSASRec in PyTorch

pytorch recommender-system sequential-models tisasrec

Last synced: 10 Nov 2024

https://github.com/flaport/photontorch

Highly parallel simulation and optimization of photonic circuits in time and frequency domain based on the deep-learning framework PyTorch

optimization photonic-circuit photonic-computing photonic-optimization photonics pytorch simulation simulation-framework simulator torch

Last synced: 06 Nov 2024

https://github.com/danfeix/rpn

Code Repository for Regression Planning Networks

machine-learning pybullet pytorch robotics robotics-simulation

Last synced: 12 Nov 2024

https://github.com/amanpriyanshu/deep-belief-networks-in-pytorch

The aim of this repository is to create RBMs, EBMs and DBNs in generalized manner, so as to allow modification and variation in model types.

dbn dbnet deep-belief-network deep-learning ebm energy-model machine-learning pytorch rbm rbm-dbn

Last synced: 18 Nov 2024

https://github.com/k-h-ismail/dilated-convolution-with-learnable-spacings-pytorch

[ICLR 2023] "Dilated convolution with learnable spacings" Ismail Khalfaoui Hassani, Thomas Pellegrini and Timothée Masquelier

pytorch

Last synced: 11 Oct 2024

https://github.com/warner-benjamin/optimi

Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers

deep-learning optimizers pytorch

Last synced: 13 Nov 2024

https://github.com/youngfly11/LCMCG-PyTorch

AAAI2020-The official implementation of "Learning Cross-modal Context Graph for Visual Grounding"

aaai2020 pytorch

Last synced: 20 Nov 2024

https://github.com/antonior92/ecg-age-prediction

Scripts and modules for training and testing neural network for age prediction from the ECG. Companion code to the paper "Deep neural network-estimated electrocardiographic age as a mortality predictor".

age-prediction biological-age convolutional-neural-networks deep-learning ecg ecg-records ecg-signal machine-learning neural-networks pytorch

Last synced: 16 Oct 2024

https://github.com/rejuvyesh/pycallchainrules.jl

Differentiate python calls from Julia

autodifferentiation jax julia pytorch

Last synced: 28 Oct 2024

https://github.com/Toytiny/RaFlow

Self-Supervised Scene Flow Estimation with 4-D Automotive Radar (RA-L & IROS'22)

automotive-radar autonomous-driving deep-learning point-cloud pytorch scene-flow-estimation self-supervised

Last synced: 28 Oct 2024

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

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

alpaca chatgpt deepspeed finetune gpt llama llm lora peft ppo pytorch reward-models rlhf

Last synced: 11 Nov 2024

https://github.com/lab-cosmo/sphericart

Multi-language library for the calculation of spherical harmonics in Cartesian coordinates

geometric-deep-learning python pytorch spherical-harmonics

Last synced: 13 Nov 2024

https://github.com/lonepatient/multi-sample_dropout_pytorch

a simple pytorch implement of Multi-Sample Dropout

cifar10 loss multi-sample-dropout pytorch resnet

Last synced: 06 Nov 2024

https://github.com/unixwzrd/oobabooga-macOS

Optimizing performance, building and installing packages required for oobabooga, AI and Data Science on Apple Silicon GPU. The goal is to optimize wherever possible, from the ground up.

blas journal llama-cpp-python macos numpy oobabooga pytorch

Last synced: 09 Aug 2024

https://github.com/zekeriyyaa/maskrcnn-modanet-fashion-segmentation-and-classification

Using modanet fashion dataset, the clothes images were classified under 5 season (summer,winter,spring,autumn,all).

annotati classification computer-vision deep-learning fashion kera maskrcnn modanet model python pytorch season season-classification segmentation

Last synced: 09 Nov 2024

https://github.com/zllrunning/facetools

Easy-to-use face related tools, including face detection, landmark localization, alignment & recognition, based on PyTorch.

face-alignment face-landmark-detection face-recognition mtcnn pytorch

Last synced: 08 Nov 2024

https://github.com/vita-group/self-pu

[ICML2020] "Self-PU: Self Boosted and Calibrated Positive-Unlabeled Training" by Xuxi Chen, Wuyang Chen, Tianlong Chen, Ye Yuan, Chen Gong, Kewei Chen, Zhangyang Wang

pu-learning pytorch

Last synced: 16 Nov 2024

https://github.com/hkproj/pytorch-lora

LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch

lora low-rank-adaptation paper-implementations pytorch

Last synced: 17 Nov 2024

https://gitlab.com/EAVISE/lightnet

Darknet got illuminated by PyTorch ~ Meet Lightnet

CNN computer vision darknet deep learning object detection pytorch single shot detection yolo

Last synced: 27 Oct 2024

https://github.com/vra/dinov2-retrieval

A cli program of image retrieval using dinov2

ai computer-vision deep-learning dinov2 image-retrieval pytorch

Last synced: 07 Nov 2024

https://github.com/d-li14/lambda.pytorch

PyTorch implementation of Lambda Network and pretrained Lambda-ResNet

attention iclr2021 imagenet lambda-network pre-trained-model pytorch

Last synced: 07 Nov 2024

https://github.com/aangelopoulos/conformal-risk

Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vision and natural language processing.

computer-vision conformal conformal-prediction natural-language-processing python pytorch pytorch-implementation uncertainty-estimation uncertainty-quantification

Last synced: 05 Nov 2024

https://github.com/toshas/torch_truncnorm

Truncated Normal Distribution in PyTorch

distributions pytorch truncated-normal

Last synced: 16 Nov 2024

https://github.com/tomrunia/pytorchsteerablepyramid

PyTorch implementation of the Complex Steerable Pyramid

batch computer-vision cuda image-processing mkl pyramid pytorch

Last synced: 13 Nov 2024

https://github.com/drscotthawley/aeiou

(ML) audio engineering i/o utils

audio machine-learning pytorch utilities

Last synced: 18 Nov 2024

https://github.com/xinario/catgan_pytorch

Unsupervised and Semi-supervised Learning with Categorical Generative Adversarial Networks

catgan dcgan gan generative-adversarial-network pytorch semi-supervised-learning unsupervised-learning

Last synced: 10 Nov 2024

https://github.com/astorfi/cor-gan

:unlock: COR-GAN: Correlation-Capturing Convolutional Neural Networks for Generating Synthetic Healthcare Records

deep-learning generative-adversarial-network healthcare python pytorch synthetic-dataset-generation

Last synced: 22 Oct 2024

https://github.com/dl4to/dl4to

DL4TO is a Python library for 3D topology optimization that is based on PyTorch and allows easy integration with neural networks.

3d deep-learning neural-network neural-networks pytorch topology-optimization

Last synced: 08 Nov 2024

https://github.com/bshall/vectorquantizedvae

A PyTorch implementation of "Continuous Relaxation Training of Discrete Latent Variable Image Models"

generative-models pytorch vae vq-vae

Last synced: 13 Nov 2024

https://github.com/OIdiotLin/torchtracer

A python package for visualization and storage management in a pytorch AI task.

lab machine-learning pytorch tools

Last synced: 16 Nov 2024

https://github.com/lonepatient/ernie-text-classification-pytorch

This repo contains a PyTorch implementation of a pretrained ERNIE model for text classification.

bert chinese-text-classification ernie python3 pytorch text-classification

Last synced: 06 Nov 2024

https://github.com/harry24k/cw-pytorch

A pytorch implementation of "Towards Evaluating the Robustness of Neural Networks"

adversarial-attacks deep-learning pytorch

Last synced: 06 Nov 2024

https://github.com/zjysteven/dverge

Pytorch implementation of our NeurIPS'20 *Oral* paper "DVERGE: Diversifying Vulnerabilities for Enhanced Robust Generation of Ensembles".

adversarial-attack-and-defense adversarial-robustness diversifying-vulnerabilities ensembles pytorch

Last synced: 02 Nov 2024

https://github.com/gabrielepicco/deep-learning-flower-identifier

Deep learning network to identify 102 different types of flowers

google-colaboratory image-classification pytorch pytorch-scholarship-challenge

Last synced: 08 Nov 2024

https://github.com/byminji/SLTtrack

Official Implementation of Towards Sequence-Level Training for Visual Tracking (ECCV 2022)

eccv2022 pytorch reinforcement-learning single-object-tracking sot tracking visual-tracking vot

Last synced: 02 Nov 2024

https://github.com/crowsonkb/vgg_loss

A VGG-based perceptual loss function for PyTorch.

autoencoder perceptual-losses pytorch vgg

Last synced: 14 Oct 2024

https://github.com/xierc/Semi_Human_Pose

[ICCV-2021] An Empirical Study of the Collapsing Problem in Semi-Supervised 2D Human Pose Estimation

consistency-regularization human-pose-estimation pytorch semi-supervised-learning

Last synced: 08 Aug 2024