vmoens Goto Github PK

aafig

Sphinx extension to add ASCII-art diagrams using aafigure

accelerate

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

adafvf.jl

Julia/Flux.jl implementation of the AdaFVF, a Bayesian version of Adam inspired by Stabilised Forgetting algorithms

autocat

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

dm_control

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

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.

functorch

functorch is JAX-like composable function transforms for PyTorch.

graphviz

Simple Python interface for Graphviz

gym

A toolkit for developing and comparing reinforcement learning algorithms.

gymnasium

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

habitat-lab

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

habitat-sim

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

lerobot

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

mj_envs

A collection of MuJoCo based environments.

mjrl

Reinforcement learning algorithms for MuJoCo tasks

mujoco

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

mujoco-py

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

pyraedm

PyTorch version of the ReaDm

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

r3m

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

raedm.jl

Julia/Flux.jl implementation of the Recurrent auto-encoding Diffusion model

rl

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

rtd-tutorial

tensordict

TensorDict is a pytorch dedicated tensor container.

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.

theseus

A library for differentiable nonlinear optimization

torchrec

Pytorch domain library for recommendation systems

torchrl_examples

Training examples using TorchRL repo

torchrl_mcts

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

torchsnapshot

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