vmoens

vmoens's Stars

• meta-llama/llama

  Inference code for Llama models

  Language:Python58k5351.1k9.7k

• pytorch/vision

  Datasets, Transforms and Models specific to Computer Vision

  Language:Python16.7k4663.4k7k

• NicolasHug/Surprise

  A Python scikit for building and analyzing recommender systems

  Language:Python6.6k1423851k

• pytorch/torchtune

  PyTorch native post-training library

  Language:Python5.1k47954569

• MarkFzp/mobile-aloha

  Mobile ALOHA: Learning Bimanual Mobile Manipulation with Low-Cost Whole-Body Teleoperation

  Language:Jupyter Notebook4.1k7518695

• docarray/docarray

  Represent, send, store and search multimodal data

  Language:Python3k45641233

• facebookresearch/habitat-lab

  A modular high-level library to train embodied AI agents across a variety of tasks and environments.

  Language:Python2.3k46704538

• pytorch/ao

  PyTorch native quantization and sparsity for training and inference

  Language:Python1.9k43435235

• PufferAI/PufferLib

  Simplifying reinforcement learning for complex game environments

  Language:C1.9k1620109

• facebookresearch/theseus

  A library for differentiable nonlinear optimization

  Language:Python1.9k32195132

• NVIDIA-Omniverse/IsaacGymEnvs

  Isaac Gym Reinforcement Learning Environments

  Language:Python1.7k36189375

• facebookincubator/submitit

  Python 3.8+ toolbox for submitting jobs to Slurm

  Language:Python1.4k23123134

• facebookresearch/torchbeast

  A PyTorch Platform for Distributed RL

  Language:Python7471537115

• NVIDIA-Omniverse/OmniIsaacGymEnvs

  Reinforcement Learning Environments for Omniverse Isaac Gym

  Language:Python73017158200

• ai4co/rl4co

  A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

  Language:Python54889995

• rail-berkeley/serl

  SERL: A Software Suite for Sample-Efficient Robotic Reinforcement Learning

  Language:Python505113057

• facebookresearch/eai-vc

  The repository for the largest and most comprehensive empirical study of visual foundation models for Embodied AI (EAI).

  Language:Python480201944

• araffin/sbx

  SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms

  Language:Python428153138

• NVIDIA-Merlin/dataloader

  The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX

  Language:Python417173926

• proroklab/VectorizedMultiAgentSimulator

  VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of a vectorized 2D physics engine written in PyTorch and a set of challenging multi-robot scenarios. Additional scenarios can be implemented through a simple and modular interface.

  Language:Python40676279

• Eclectic-Sheep/sheeprl

  Distributed Reinforcement Learning accelerated by Lightning Fabric

  Language:Python359412144

• sanjeevanahilan/nanoChatGPT

  A crude RLHF layer on top of nanoGPT with Gumbel-Softmax trick

  Language:Python2899025

• pytorch-labs/float8_experimental

  This repository contains the experimental PyTorch native float8 training UX

  Language:Python222254719

• pytorch/test-infra

  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

  Language:TypeScript91173.1k90

• Acellera/acegen-open

  Language models for drug discovery using torchrl

  Language:Python7941113

• moratodpg/imp_marl

  IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL

  Language:Python41556

• facebookresearch/gen_dgrl

  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024

  Language:Python28315

• compsciencelab/torchrl_examples

  Training examples using TorchRL repo

  Language:Python7302

• SchweizerischeBundesbahnen/flatland-torchrl

  An adaption of the Flatland environment for TorchRL.

  Language:Python520

• kurt-stolle/unipercept

  Various libraries and utilities for rapid computer vision research in PyTorch

  Language:Jupyter Notebook4400