Skip to main content

A library for Reinforcement Learning

Project description

# anyrl-py

This is a Python remake (and makeover) of [anyrl](https://github.com/unixpickle/anyrl). It is a general-purpose library for Reinforcement Learning which aims to be as modular as possible.

# Installation

You can install anyrl with pip:

` pip install anyrl `

# APIs

There are several different sub-modules in anyrl:

  • models: abstractions and concrete implementations of RL models. This includes actor-critic RNNs, MLPs, CNNs, etc. Takes care of sequence padding, BPTT, etc.

  • envs: APIs for dealing with environments, including wrappers and asynchronous environments.

  • rollouts: APIs for gathering and manipulating batches of episodes or partial episodes. Many RL algorithms include a “gather trajectories” step, and this sub-module fulfills that role.

  • algos: well-known learning algorithms like policy gradients or PPO. Also includes mini-algorithms like Generalized Advantage Estimation.

  • spaces: tools for using action and observation spaces. Includes parameterized probability distributions for implementing stochastic policies.

# Motivation

Currently, most RL code out there is very restricted and not properly decoupled. In contrast, anyrl aims to be extremely modular and flexible. The goal is to decouple agents, learning algorithms, trajectories, and things like GAE.

For example, anyrl decouples rollouts from the learning algorithm (when possible). This way, you can gather rollouts in several different ways and still feed the results into one learning algorithm. Further, and more obviously, you don’t have to rewrite rollout code for every new RL algorithm you implement. However, algorithms like A3C and Evolution Strategies may have specific ways of performing rollouts that can’t rely on the rollout API.

# Use of TensorFlow

This project relies on TensorFlow for models and training algorithms. However, anyrl APIs are framework-agnostic when possible. For example, the rollout API can be used with any policy, whether it’s a TensorFlow neural network or a native-Python decision forest.

# TODO

Here is the current TODO list, organized by sub-module:

  • models * Unify CNN and MLP models with a single base class. * Unshared actor-critics for TRPO and the like.

  • rollouts * Maybe: way to not record states in model_outs (memory saving) * Normalization based on advantage magnitudes. * Optimize for sub-batches in BatchedPlayer.

  • algos * TRPO * PPO: allow clipping for value function

  • spaces * Dict

  • tests * Benchmarks for rollouts * Benchmarks for training

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

anyrl-0.11.16.tar.gz (56.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

anyrl-0.11.16-py3-none-any.whl (80.9 kB view details)

Uploaded Python 3

File details

Details for the file anyrl-0.11.16.tar.gz.

File metadata

  • Download URL: anyrl-0.11.16.tar.gz
  • Upload date:
  • Size: 56.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No

File hashes

Hashes for anyrl-0.11.16.tar.gz
Algorithm Hash digest
SHA256 f7f2cf6b96002197d9c88be7bffc1f13df9a5c8ada81973a7058fbb5fcb41726
MD5 1aef0875916caa08336a4940884ac13b
BLAKE2b-256 16e4116d06f969ebef4f7e79362d633ca9a5f087b71ede1598665b5489035cbc

See more details on using hashes here.

File details

Details for the file anyrl-0.11.16-py3-none-any.whl.

File metadata

File hashes

Hashes for anyrl-0.11.16-py3-none-any.whl
Algorithm Hash digest
SHA256 e33591758de1a7f2a21524945bfc1a5c16c5dc3a81861cef47140f7b0bdc821d
MD5 31ea5df398d03118c4db482fc7bbfceb
BLAKE2b-256 04aa009b2ca0e2ce925673ee76fe90cd239df5534fd6d6b73e1b45afed0c11c0

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page