YN35

YN35's Stars

• buoyancy99/diffusion-forcing

  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"

  Language:Python49519

• hill-a/stable-baselines

  A fork of OpenAI Baselines, implementations of reinforcement learning algorithms

  Language:Python4.1k724

• dennybritz/reinforcement-learning

  Implementation of Reinforcement Learning Algorithms. Python, OpenAI Gym, Tensorflow. Exercises and Solutions to accompany Sutton's Book and David Silver's course.

  Language:Jupyter Notebook20.4k6k

• MaxHalford/pytorch-resample

  🎲 Iterable dataset resampling in PyTorch

  Language:Python915

• rainbow979/robodreamer

  Language:Python351

• nyuolab/VIPER-torch

  dreamerv3 in pytorch

  Language:Python2

• matsuolab/BREMEN

  Codebase of Deployment-Efficient Reinforcement Learning via Model-Based Offline Optimization (ICLR2021)

  Language:Python517

• facebookresearch/DiT

  Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

  Language:Python6k534

• JuanBindez/pytubefix

  Python3 library for downloading YouTube Videos.

  Language:Python45467

• DigiRL-agent/digirl

  Official repo for paper DigiRL: Training In-The-Wild Device-Control Agents with Autonomous Reinforcement Learning.

  Language:Python20016

• PixArt-alpha/PixArt-alpha

  PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis

  Language:Python2.7k168

• zugexiaodui/torch_flops

  A library for calculating the FLOPs in the forward() process based on torch.fx

  Language:Python712

• facebookresearch/fvcore

  Collection of common code that's shared among different research projects in FAIR computer vision team.

  Language:Python2k226

• MrYxJ/calculate-flops.pytorch

  The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)

  Language:Python49016

• NVIDIA/TransformerEngine

  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

  Language:Python1.8k300

• webdataset/webdataset

  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

  Language:Python2.2k174

• omni-us/jsonargparse

  Implement minimal boilerplate CLIs derived from type hints and parse from command line, config files and environment variables

  Language:Python31644

• microsoft/debugpy

  An implementation of the Debug Adapter Protocol for Python

  Language:Python1.8k131

• facebookresearch/Pearl

  A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.

  Language:Jupyter Notebook2.6k154

• madebyollin/taesd

  Tiny AutoEncoder for Stable Diffusion

  Language:Python53227

• lucidrains/magvit2-pytorch

  Implementation of MagViT2 Tokenizer in Pytorch

  Language:Python53735

• dwjshift/IL_ADS

  code for the paper Imitation Learning from Observation with Automatic Discount Scheduling

  Language:Python81

• argmax-ai/aime-v2

  Official implementation of AIME-v2 from "Overcoming Knowledge Barriers: Online Imitation Learning from Observation with Pretrained World Models"

  Language:Python3

• hpcaitech/Open-Sora

  Open-Sora: Democratizing Efficient Video Production for All

  Language:Python21.6k2.1k

• clvrai/goal_prox_il

  Generalizable Imitation Learning from Observation via Inferring Goal Proximity (NeurIPS 2021)

  Language:Python221

• medric49/gaifo

  Generative adversarial Imitation from Observation

  Language:Python1

• facebookresearch/segment-anything-2

  The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

  Language:Jupyter Notebook10.7k862

• lucidrains/vector-quantize-pytorch

  Vector (and Scalar) Quantization, in Pytorch

  Language:Python2.4k196

• myscience/open-genie

  Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).

  Language:Python494

• YongfeiYan/Gumbel_Softmax_VAE

  PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution

  Language:Python19837