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.

GitHub: https://github.com/topics/pytorch
Wikipedia: https://en.wikipedia.org/wiki/PyTorch
Repo: https://github.com/pytorch/pytorch
Created by: Facebook's AI Research lab (FAIR)
Released: September 2016
Related Topics: python, pytorch-tutorial,
Last updated: 2024-07-29 13:55:27 UTC
JSON Representation

https://github.com/DanAnastasyev/DeepNLP-Course

Deep NLP Course

colab-notebook deep-learning keras nlp pytorch

Last synced: 04 Aug 2024

https://github.com/grok-ai/nn-template

Generic template to bootstrap your PyTorch project.

best-practices best-practises cookiecutter deep-learning dvc github-actions huggingface-datasets hydra mkdocs pre-commit project-structure pytorch pytorch-lightning reproducibility research streamlit template wandb weights-and-biases

Last synced: 01 Aug 2024

https://github.com/YU1ut/MixMatch-pytorch

Code for "MixMatch - A Holistic Approach to Semi-Supervised Learning"

deep-learning pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/sunshineatnoon/Paper-Implementations

Use PyTorch to implement some classic frameworks

computer-vision deep-learning pytorch

Last synced: 07 Aug 2024

https://github.com/SamsungLabs/ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

hrnets interactive-segmentation pretrained-models pytorch segmentation

Last synced: 03 Aug 2024

https://github.com/wanmeihuali/taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.

3d-reconstruction 3d-rendering computer-graphics computer-vision machine-learning nerf python pytorch real-time-rendering taichi

Last synced: 01 Aug 2024

https://github.com/ternaus/robot-surgery-segmentation

Wining solution and its improvement for MICCAI 2017 Robotic Instrument Segmentation Sub-Challenge

computer-vision deep-learning image-segmentation medical-imaging python pytorch robot-assisted-surgery

Last synced: 07 Aug 2024

https://github.com/lcswillems/rl-starter-files

RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code

a2c a3c minigrid multi-process ppo preprocessed-observations pytorch reward-shaping

Last synced: 07 Aug 2024

https://github.com/Ha0Tang/AttentionGAN

AttentionGAN for Unpaired Image-to-Image Translation & Multi-Domain Image-to-Image Translation

adversarial-networks attention-model computer-vision cyclegan deep-learning gans geometry ijcnn image-generation image-to-image-translation image-translation multi-domain pytorch unpaired

Last synced: 03 Aug 2024

https://github.com/chrischoy/FCGF

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor

Last synced: 01 Aug 2024

https://github.com/ymcui/MacBERT

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

bert language-model macbert nlp pytorch tensorflow transformers

Last synced: 03 Aug 2024

https://github.com/cdqa-suite/cdQA

⛔ [NOT MAINTAINED] An End-To-End Closed Domain Question Answering System.

artificial-intelligence bert deep-learning information-retrieval natural-language-processing nlp pytorch question-answering reading-comprehension transformers

Last synced: 02 Aug 2024

https://github.com/alankbi/detecto

Build fully-functioning computer vision models with PyTorch

computer-vision faster-rcnn machine-learning object-detection python pytorch

Last synced: 30 Jul 2024

https://github.com/lucasjinreal/DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

centernet dcnv2 pytorch

Last synced: 31 Jul 2024

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

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

hydra pytorch pytorch-lightning transformers

Last synced: 01 Aug 2024

https://github.com/plemeri/transparent-background

This is a background removing tool powered by InSPyReNet (ACCV 2022)

background-removal deep-learning dichotomous-image-segmentation image-matting image-processing photo-editing python pytorch remove-background remove-background-image remove-background-video salient-object-detection video-editing

Last synced: 01 Aug 2024

https://github.com/OndrejTexler/Few-Shot-Patch-Based-Training

The official implementation of our SIGGRAPH 2020 paper Interactive Video Stylization Using Few-Shot Patch-Based Training

deep-learning gan generative-adversarial-network image-generation image-to-image-translation pytorch style-transfer

Last synced: 03 Aug 2024

https://github.com/VincentStimper/normalizing-flows

PyTorch implementation of normalizing flow models

density-estimation glow invertible-neural-networks neural-spline-flow normalizing-flow pytorch real-nvp residual-flow variational-autoencoder variational-inference

Last synced: 03 Aug 2024

https://github.com/BlackSamorez/tensor_parallel

Automatically split your PyTorch models on multiple GPUs for training & inference

deep-learning machine-learning natural-language-processing nlp python pytorch pytorch-transformers

Last synced: 09 Aug 2024

https://github.com/yanx27/EverybodyDanceNow_reproduce_pytorch

Everybody dance now reproduced in pytorch

everybody-dance-now openpose pix2pixhd pose2pose pytorch

Last synced: 01 Aug 2024

https://github.com/lucastabelini/LaneATT

Code for the paper entitled "Keep your Eyes on the Lane: Real-time Attention-guided Lane Detection" (CVPR 2021)

computer-vision deep-learning lane-detection pytorch

Last synced: 31 Jul 2024

https://github.com/nianticlabs/manydepth

[CVPR 2021] Self-supervised depth estimation from short sequences

cityscapes cost-volumes cvpr cvpr2021 depth-estimation depths estimating-depths kitti monodepth pytorch self-supervised self-supervised-learning

Last synced: 05 Aug 2024

https://github.com/alvinwan/neural-backed-decision-trees

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet

cifar10 cifar100 decision-trees explainability image-classification imagenet interpretability neural-backed-decision-trees neural-networks pretrained-models pretrained-weights pytorch tiny-imagenet

Last synced: 03 Aug 2024

https://github.com/castorini/daam

Diffusion attentive attribution maps for interpreting Stable Diffusion.

diffusion explainable-ai generative-ai huggingface pytorch stable-diffusion

Last synced: 03 Aug 2024

https://github.com/signatrix/efficientdet

(Pretrained weights provided) EfficientDet: Scalable and Efficient Object Detection implementation by Signatrix GmbH

coco-dataset deep-learning efficientdet efficientnet neural-network object-detection pytorch

Last synced: 07 Aug 2024

https://github.com/mryab/efficient-dl-systems

Efficient Deep Learning Systems course materials (HSE, YSDA)

cuda deep-learning distributed-training efficient-deep-learning machine-learning ml-infrastructure mlops pytorch

Last synced: 01 Aug 2024

https://github.com/JiahuiYu/wdsr_ntire2018

Code of our winning entry to NTIRE super-resolution challenge, CVPR 2018

deep-neural-networks efficient-algorithm pytorch super-resolution wdsr

Last synced: 07 Aug 2024

https://github.com/mit-han-lab/lite-transformer

[ICLR 2020] Lite Transformer with Long-Short Range Attention

nlp pytorch transformer

Last synced: 03 Aug 2024

https://github.com/abeardear/pytorch-YOLO-v1

an experiment for yolo-v1, including training and testing.

experiment object-detection pytorch yolov1

Last synced: 02 Aug 2024

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

PyTorch Boilerplate For Research

pytorch segmentation

Last synced: 02 Aug 2024

https://github.com/araffin/robotics-rl-srl

S-RL Toolbox: Reinforcement Learning (RL) and State Representation Learning (SRL) for Robotics

arm baselines baxter-robot gym kuka pybullet pytorch reinforcement-learning representation-learning robotics state

Last synced: 31 Jul 2024

https://github.com/augmentedstartups/AS-One

Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.

computer-vision deep-learning object-detection opencv pytorch sam tracking ultralytics yolor yolov5 yolov7 yolov8 yolov9 yolox

Last synced: 02 Aug 2024

https://github.com/richzhang/colorization-pytorch

PyTorch reimplementation of Interactive Deep Colorization

computer-graphics computer-vision convolutional-neural-networks deep-learning image-colorization pytorch siggraph

Last synced: 03 Aug 2024

https://github.com/facebookresearch/fairseq2

FAIR Sequence Modeling Toolkit 2

artificial-intelligence deep-learning machine-learning python pytorch

Last synced: 02 Aug 2024

https://github.com/hirofumi0810/neural_sp

End-to-end ASR/LM implementation with PyTorch

asr attention attention-mechanism automatic-speech-recognition ctc language-model language-modeling pytorch rnn-transducer seq2seq sequence-to-sequence speech speech-recognition streaming transformer transformer-xl

Last synced: 02 Aug 2024

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

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch rough-paths time-series

Last synced: 01 Aug 2024

https://github.com/iMoonLab/DeepHypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 31 Jul 2024

https://github.com/castorini/hedwig

PyTorch deep learning models for document classification

deep-learning document-classification pytorch

Last synced: 01 Aug 2024

https://github.com/Malllabiisc/CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks

deep-learning graph-convolutional-networks graph-representation-learning iclr2020 link-prediction pytorch relation-embeddings

Last synced: 08 Aug 2024

https://github.com/JackWoo0831/Yolov7-tracker

Yolo X, v7, v8 and several Multi-Object Tracker(SORT, DeepSORT, ByteTrack, BoT-SORT, etc.) in MOT17 and VisDrone2019 Dataset. It uses a unified style and integrated tracker for easy embedding in your own projects.

pytorch

Last synced: 31 Jul 2024

https://github.com/szagoruyko/functional-zoo

PyTorch and Tensorflow functional model definitions

pretrained-models pytorch tensorflow

Last synced: 03 Aug 2024

https://github.com/hrnbot/Basic-Mathematics-for-Machine-Learning

The motive behind Creating this repo is to feel the fear of mathematics and do what ever you want to do in Machine Learning , Deep Learning and other fields of AI

ai algebra beginner calculus linear-algebra machine-learning machinelearning-python mathematical-analysis mathematical-functions mathematics maths notebook prerequisites probability python pytorch siraj siraj-raval statistics

Last synced: 02 Aug 2024

https://github.com/QUVA-Lab/e2cnn

E(2)-Equivariant CNNs Library for Pytorch

cnns equivariance equivariant-network group-convolution pytorch

Last synced: 09 Aug 2024

https://github.com/bowenc0221/panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)

bottom-up cityscapes cvpr2020 deeplab detectron2 instance-segmentation panoptic-segmentation pytorch semantic-segmentation sementation

Last synced: 07 Aug 2024

https://github.com/as-ideas/ForwardTacotron

⏩ Generating speech in a single forward pass without any attention!

axelspringerai deep-learning forwardtacotron python pytorch tacotron text-to-speech tts

Last synced: 01 Aug 2024

https://github.com/alibaba/Megatron-LLaMA

Best practice for training LLaMA models in Megatron-LM

deepspeed distributed-training llama llm megatron-lm pretraining pytorch

Last synced: 31 Jul 2024

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

[ECCV 2020] Searching Efficient 3D Architectures with Sparse Point-Voxel Convolution

3d-deep-learning architecture-search computer-vision deep-learning efficiency lidar point-cloud pytorch semantickitti

Last synced: 31 Jul 2024

https://github.com/graykode/xlnet-Pytorch

Simple XLNet implementation with Pytorch Wrapper

bert natural-language-processing nlp pytorch xlnet xlnet-pytorch

Last synced: 01 Aug 2024

https://github.com/awarebayes/RecNN

Reinforced Recommendation toolkit built around pytorch 1.7

library news-recommendation pytorch pytorch-rl recommendation-system recommender-system reinforcement-learning toolkit

Last synced: 08 Aug 2024

https://github.com/songdejia/EAST

This is a pytorch re-implementation of EAST: An Efficient and Accurate Scene Text Detector.

deeplearning east icdar ocr pytorch textdetection

Last synced: 01 Aug 2024

https://github.com/HarryVolek/PyTorch_Speaker_Verification

PyTorch implementation of "Generalized End-to-End Loss for Speaker Verification" by Wan, Li et al.

pytorch speaker-identification speaker-verification

Last synced: 02 Aug 2024

https://github.com/ShuangXieIrene/ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.

darknet fssd mobilenet pytorch rfb ssd vgg yolo

Last synced: 03 Aug 2024

https://github.com/szymonmaszke/torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

deep-learning framework inference keras layers library machine-learning pytorch shape sota

Last synced: 02 Aug 2024

https://github.com/mir-group/nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

atomistic-simulations computational-chemistry deep-learning drug-discovery force-fields interatomic-potentials machine-learning materials-science molecular-dynamics pytorch

Last synced: 02 Aug 2024

https://github.com/hassony2/manopth

MANO layer for PyTorch, generating hand meshes as a differentiable layer

hand layer mano pytorch

Last synced: 17 Aug 2024

https://github.com/Confusezius/Deep-Metric-Learning-Baselines

PyTorch Implementation for Deep Metric Learning Pipelines

cars196 computer-vision cub200 deep-learning deep-metric-learning distance-sampling metric-learning neural-networks pku-vehicle pytorch shop-clothes

Last synced: 02 Aug 2024

https://github.com/ikostrikov/pytorch-flows

PyTorch implementations of algorithms for density estimation

deep-learning density-estimation neural-networks probabilities pytorch

Last synced: 30 Jul 2024

https://github.com/loudinthecloud/pytorch-ntm

Neural Turing Machines (NTM) - PyTorch Implementation

lstm neural-network neural-turing-machines notebook ntm python pytorch

Last synced: 07 Aug 2024

https://github.com/foolwood/pytorch-slimming

Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.

deep-learning fast-inference l1-regularization pytorch weight-pruning

Last synced: 03 Aug 2024

https://github.com/Hawkeye-FineGrained/Hawkeye

Open source deep learning based fine-grained image recognition toolbox built on PyTorch🔥

deep-learning fine-grained pytorch toolbox

Last synced: 01 Aug 2024

https://github.com/Scalsol/mega.pytorch

Memory Enhanced Global-Local Aggregation for Video Object Detection, CVPR2020

pytorch video-object-detection

Last synced: 02 Aug 2024

https://github.com/underneathall/pinferencia

Python + Inference - Model Deployment library in Python. Simplest model inference server ever.

ai artificial-intelligence computer-vision data-science deep-learning huggingface inference inference-server machine-learning model-deployment model-serving modelserver nlp paddlepaddle predict python pytorch serving tensorflow transformers

Last synced: 03 Aug 2024

https://github.com/GRAAL-Research/poutyne

A simplified framework and utilities for PyTorch

data-science deep-learning keras machine-learning neural-network python pytorch

Last synced: 31 Jul 2024

https://github.com/mackelab/sbi

Simulation-based inference toolkit

bayesian-inference likelihood-free-inference machine-learning parameter-estimation pytorch simulation-based-inference

Last synced: 24 Aug 2024

https://github.com/bryanyzhu/two-stream-pytorch

PyTorch implementation of two-stream networks for video action recognition

action-recognition pytorch two-stream video

Last synced: 04 Aug 2024

https://github.com/benedekrozemberczki/GraphWaveletNeuralNetwork

A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019)

convolutional-neural-networks deep-learning deepwalk gcn gpt2 gpt3 graph-convolution graph-convolutional-network graph-neural-networks graph-representation graphsage gwnn iclr machine-learning node2vec pytorch research sklearn tensorflow transformer

Last synced: 01 Aug 2024

https://github.com/megvii-model/MOTR

[ECCV2022] MOTR: End-to-End Multiple-Object Tracking with TRansformer

end-to-end multi-object-tracking pytorch transformer

Last synced: 02 Aug 2024

https://github.com/megvii-research/MOTR

[ECCV2022] MOTR: End-to-End Multiple-Object Tracking with TRansformer

end-to-end multi-object-tracking pytorch transformer

Last synced: 31 Jul 2024

https://github.com/ignacio-rocco/detectorch

Detectorch - detectron for PyTorch

detectron instance-segmentation object-detection object-segmentation python pytorch

Last synced: 03 Aug 2024

https://github.com/AviSoori1x/makeMoE

From scratch implementation of a sparse mixture of experts language model inspired by Andrej Karpathy's makemore :)

deep-learning large-language-models llm mixture-of-experts neural-networks pytorch pytorch-implementation

Last synced: 01 Aug 2024

https://github.com/longcw/RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch

pytorch

Last synced: 03 Aug 2024

https://github.com/sbi-dev/sbi

Simulation-based inference toolkit

bayesian-inference likelihood-free-inference machine-learning parameter-estimation pytorch simulation-based-inference

Last synced: 02 Aug 2024

https://github.com/hpcaitech/FastFold

Optimizing AlphaFold Training and Inference on GPU Clusters

alphafold2 cuda evoformer gpu habana-gaudi parallelism protein-folding protein-structure pytorch

Last synced: 02 Aug 2024

https://github.com/RobotLocomotion/pytorch-dense-correspondence

Code for "Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulation"

3d artificial-intelligence computer-vision deep-learning manipulation pytorch robotics self-supervised-learning vision

Last synced: 03 Aug 2024

https://github.com/huggingface/transformers-bloom-inference

Fast Inference Solutions for BLOOM

bloom huggingface-transformers nlp pytorch

Last synced: 03 Aug 2024

https://github.com/xwying/torchshow

Visualize PyTorch tensors with a single line of code.

image-tensor notebook pytorch tensor visualization

Last synced: 02 Aug 2024

https://github.com/KDD-OpenSource/DeepADoTS

Repository of the paper "A Systematic Evaluation of Deep Anomaly Detection Methods for Time Series".

anomaly-detection deep-learning pytorch tensorflow time-series timeseries

Last synced: 30 Jul 2024

https://github.com/stochasticai/x-stable-diffusion

Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention. Join our Discord communty: https://discord.com/invite/TgHXuSJEk6

aitemplate automl cuda docker inference notebook nvfuser onnx onnxruntime pytorch stable-diffusion tensorrt

Last synced: 01 Aug 2024

https://github.com/ternaus/TernausNetV2

TernausNetV2: Fully Convolutional Network for Instance Segmentation

computer-vision deep-learning image-segmentation python pytorch satellite-imagery

Last synced: 04 Aug 2024

https://github.com/AmazingDD/daisyRec

Official code for "DaisyRec 2.0: Benchmarking Recommendation for Rigorous Evaluation" (TPAMI2022) and "Are We Evaluating Rigorously? Benchmarking Recommendation for Reproducible Evaluation and Fair Comparison" (RecSys2020)

afm biasmf cdae collaborative-filtering deepfm ease factorization-machines item2vec k-nearest-neighbors matrix-factorization neumf neural-collaborative-filtering nfm pytorch recommender-system slim svdpp vae

Last synced: 08 Aug 2024

https://github.com/NoCLin/LightMirrors

LightMirrors is a lightweight mirror server with caching capabilities that currently supports DockerHub, K8S, PyPI, PyTorch, and NPM.

cache-proxy docker docker-compose dockerhub ghcr k8s mirror mirror-site mirrors npm nvcr pypi python pytorch quay reverse-proxy

Last synced: 31 Jul 2024

https://github.com/RelevanceAI/vectorhub

Vector Hub - Library for easy discovery, and consumption of State-of-the-art models to turn data into vectors. (text2vec, image2vec, video2vec, graph2vec, bert, inception, etc)

artificial-intelligence audio-processing deep-learning deeplearning embeddings encodings image2vec machine-learning neural-network python pytorch tensorflow tfhub transformers vector vector-similarity video-processing word2vec

Last synced: 02 Aug 2024

https://github.com/vector-ai/vectorhub

Vector Hub - Library for easy discovery, and consumption of State-of-the-art models to turn data into vectors. (text2vec, image2vec, video2vec, graph2vec, bert, inception, etc)

Last synced: 15 Aug 2024

https://github.com/achuthasubhash/Complete-Life-Cycle-of-a-Data-Science-Project

Complete-Life-Cycle-of-a-Data-Science-Project

analysis data-analysis data-science dataset deep-learning eda exploratory-data-analysis feature-engineering federated-learning machine-learning nlp-models python python-library pytorch reinforcement-learning scraper supervised-learning transfer-learning unsupervised-learning web-scraping

Last synced: 02 Aug 2024

https://github.com/ctallec/world-models

Reimplementation of World-Models (Ha and Schmidhuber 2018) in pytorch

model-based-rl pytorch reinforcement-learning

Last synced: 07 Aug 2024

https://github.com/omriav/blended-latent-diffusion

Official implementation for "Blended Latent Diffusion" [SIGGRAPH 2023]

computer-vision deep-learning diffusion diffusion-models generative-model image-generation multimodal multimodal-deep-learning pytorch text-driven-editing text-guided-manipulation text-to-image text-to-image-synthesis

Last synced: 31 Jul 2024

https://github.com/xlang-ai/UnifiedSKG

[EMNLP 2022] Unifying and multi-tasking structured knowledge grounding with language models

data-to-text fact-verification huggingface-datasets huggingface-transformers multi-task-learning natural-language-processing nlp prompt-learning pytorch question-answering semantic-parsing structured-knowledge-grounding text-generation

Last synced: 03 Aug 2024

https://github.com/csinva/csinva.github.io

Slides, paper notes, class notes, blog posts, and research on ML 📉, statistics 📊, and AI 🤖.

ai artificial-intelligence awesome blog computational-neuroscience data-science deep-learning jekyll-themes machine-learning machine-learning-tutorials ml neuroscience notes python pytorch research slides statistics stats website

Last synced: 10 Aug 2024

https://github.com/snap-stanford/deepsnap

Python library assists deep learning on graphs

deep-learning graph-neural-networks pytorch

Last synced: 02 Aug 2024

https://github.com/hendrycks/outlier-exposure

Deep Anomaly Detection with Outlier Exposure (ICLR 2019)

anomaly anomaly-detection calibration deep-learning ml-safety out-of-distribution-detection pytorch

Last synced: 13 Aug 2024

https://github.com/akanimax/pro_gan_pytorch

Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"

adversarial-machine-learning convolutional-neural-network gan progressive-growing-of-gans pytorch

Last synced: 03 Aug 2024

https://github.com/webis-de/small-text

Active Learning for Text Classification in Python

active-learning deep-learning looking-for-contributors machine-learning natural-language-processing nlp python pytorch text-classification transformers

Last synced: 02 Aug 2024

https://github.com/Langboat/Mengzi

Mengzi Pretrained Models

bert chinese-bert deep-learning language-understanding natural-language-processing nlp pytorch

Last synced: 31 Jul 2024

https://github.com/voidful/textrl

Implementation of ChatGPT RLHF (Reinforcement Learning with Human Feedback) on any generation model in huggingface's transformer (blommz-176B/bloom/gpt/bart/T5/MetaICL)

chatgpt controlled-nlg gpt-2 gpt-3 language-model nlg nlp pytorch reinforcement-learning rlhf

Last synced: 02 Aug 2024

https://github.com/voidful/TextRL

Implementation of ChatGPT RLHF (Reinforcement Learning with Human Feedback) on any generation model in huggingface's transformer (blommz-176B/bloom/gpt/bart/T5/MetaICL)

chatgpt controlled-nlg gpt-2 gpt-3 language-model nlg nlp pytorch reinforcement-learning rlhf