Stars
A modular high-level library to train embodied AI agents across a variety of tasks and environments.
Large World Model -- Modeling Text and Video with Millions Context
(Crafter + NetHack) in JAX. ICML 2024 Spotlight.
Python scripting for Godot. This GDExtension plugin allows you to use Python like GDScript, accessing the Godot engine's features and benefiting from Python's vast ecosystem.
Official code for Goldfish model for long video understanding and MiniGPT4-video for short video understanding
Ego4D Goal-Step: Toward Hierarchical Understanding of Procedural Activities (NeurIPS 2023)
BenchMARL is a library for benchmarking Multi-Agent Reinforcement Learning (MARL). BenchMARL allows to quickly compare different MARL algorithms, tasks, and models while being systematically ground…
Multi Robot Systems gym environment based on Pybullet
Official Pytorch implementation of "Graphit: A Unified Framework for Diverse Image Editing Tasks"
Convert Machine Learning Code Between Frameworks
Deep reinforcement learning framework for fast prototyping based on PyTorch
PyTorch implementation of normalizing flow models
A latent text-to-image diffusion model
A small package to create visualizations of PyTorch execution graphs
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 …
This is a minimal example to demonstrate how multi-agent reinforcement learning with differentiable communication channels and centralized critics can be realized in RLLib. This example serves as a…
Turning SymPy expressions into PyTorch modules.
gradslam is an open source differentiable dense SLAM library for PyTorch
Fine-tuned MARL algorithms on SMAC (100% win rates on most scenarios)
Python ROS depth estimation from RGB image based on code from the paper "High Quality Monocular Depth Estimation via Transfer Learning"
A library for efficient similarity search and clustering of dense vectors.
🐍 The official Python client library for Google's discovery based APIs.
Python3 script for interaction with https://fifteen.ai/
Dataset for use with Tacotron / 2 / Wavenet