nathankong's Stars
naver/mast3r
Grounding Image Matching in 3D with MASt3R
BlinkDL/RWKV-LM
RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.
BlinkDL/ChatRWKV
ChatRWKV is like ChatGPT but powered by RWKV (100% RNN) language model, and open source.
qiskit-community/qgss-2023
All things Qiskit Global Summer School 2023: Theory to Implementation - Lecture notes, Labs and Solutions
mlabonne/llm-course
Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.
facebookresearch/Ego4d
Ego4d dataset repository. Download the dataset, visualize, extract features & example usage of the dataset
robodhruv/visualnav-transformer
Official code and checkpoint release for mobile robot foundation models: GNM, ViNT, and NoMaD.
hermesdt/reinforcement-learning
bentrevett/pytorch-rl
Tutorials for reinforcement learning in PyTorch and Gym by implementing a few of the popular algorithms. [IN PROGRESS]
Stable-Baselines-Team/stable-baselines3-contrib
Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code
Toni-SM/skrl
Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab
SainsburyWellcomeCentre/NeuralPlayground
The standardised environment for the hippocampus and entorhinal cortex models
szagoruyko/pytorchviz
A small package to create visualizations of PyTorch execution graphs
openai/baselines
OpenAI Baselines: high-quality implementations of reinforcement learning algorithms
facebookresearch/habitat-sim
A flexible, high-performance 3D simulator for Embodied AI research.
ikostrikov/pytorch-a2c-ppo-acktr-gail
PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).
facebookresearch/habitat-lab
A modular high-level library to train embodied AI agents across a variety of tasks and environments.
Farama-Foundation/Gymnasium
An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)
Unity-Technologies/ml-agents
The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.
google-deepmind/dm_memorytasks
A set of 13 diverse machine-learning tasks that require memory to solve.
vwxyzjn/cleanrl
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
google-deepmind/acme
A library of reinforcement learning components and agents
facebookresearch/rlstructures
RLStructures is a library to facilitate the implementation of new reinforcement learning algorithms. It includes a library, a tutorial, and different RL algorithms provided as examples.
facebookresearch/rlmeta
RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.
tristandeleu/pytorch-meta
A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch
facebookresearch/SlowFast
PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
VPNL/fLoc
Functional localizer experiment used to define category-selective cortical regions
facebookresearch/pytorchvideo
A deep learning library for video understanding research.
mbiparva/slowfast-networks-pytorch
A PyTorch implementation of "SlowFast Networks for Video Recognition"
kenshohara/video-classification-3d-cnn-pytorch
Video classification tools using 3D ResNet