vmoens

flowtorch

This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a community of users and contributors by focusing initially on complete infra and documentation for how to use and create components.

hub

Submission to https://pytorch.org/hub/

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

vision

Datasets, Transforms and Models specific to Computer Vision

ghstack

Submit stacked diffs to GitHub on the command line

tdmpc2

Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"

torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

torchrl_mcts

This is a proof of concept on how MCTS can be implemented on top of TorchRL

vmoens/ghstack

Submit stacked diffs to GitHub on the command line

vmoens/tdmpc2

Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"

vmoens/torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

vmoens/torchrl_mcts

This is a proof of concept on how MCTS can be implemented on top of TorchRL

vmoens/accelerate

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

vmoens/AutoCAT

AutoCAT: Reinforcement Learning for Automated Exploration of Cache-Timing Attacks

vmoens/dm_control

DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

vmoens/flowtorch

This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a community of users and contributors by focusing initially on complete infra and documentation for how to use and create components.

vmoens/functorch

functorch is JAX-like composable function transforms for PyTorch.

vmoens/cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

vmoens/gym

A toolkit for developing and comparing reinforcement learning algorithms.

vmoens/Gymnasium

A standard API for reinforcement learning and a diverse set of reference environments (formerly Gym)

vmoens/habitat-lab

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

vmoens/habitat-sim

A flexible, high-performance 3D simulator for Embodied AI research.

vmoens/lerobot

🤗 LeRobot: State-of-the-art Machine Learning for Real-World Robotics in Pytorch

vmoens/mj_envs

A collection of MuJoCo based environments.

vmoens/mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.

vmoens/mujoco-py

MuJoCo is a physics engine for detailed, efficient rigid body simulations with contacts. mujoco-py allows using MuJoCo from Python 3.

vmoens/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

vmoens/r3m

Pre-training Reusable Representations for Robotic Manipulation Using Diverse Human Video Data

vmoens/rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

vmoens/tensordict

TensorDict is a pytorch dedicated tensor container.

vmoens/test-infra

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

vmoens/theseus

A library for differentiable nonlinear optimization

vmoens/torchrec

Pytorch domain library for recommendation systems

vmoens/torchrl_examples

Training examples using TorchRL repo

vmoens/torchsnapshot

A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.

vmoens/torchtune

A Native-PyTorch Library for LLM Fine-tuning

vmoens/tutorials

PyTorch tutorials.

vmoens/vmoens.github.io