fomorians-oss / pyoneer

Tensor utilities, reinforcement learning, and more!
https://pypi.org/project/fomoro-pyoneer/
Other
9 stars 2 forks source link
deep-learning eager-execution gym machine-learning reinforcement-learning tensorflow tensorflow-probability

pyoneer

Tensor utilities, reinforcement learning, and more! Designed to make research easier with low-level abstractions for common operations.

Usage

For the top-level utilities, import like so:

import pyoneer as pynr
pynr.math.rescale(...)

For the larger sub-modules, such as reinforcement learning, we recommend:

import pyoneer.rl as pyrl
loss_fn = pyrl.losses.PolicyGradient(...)

In general, the Pyoneer API tries to adhere to the TensorFlow 2.0 API.

Examples

API

Activations (pynr.activations)

Debugging (pynr.debugging)

Distributions (pynr.distributions)

Initializers (pynr.initializers)

Layers (pynr.layers)

Tensor Manipulation (pynr.manip)

Math (pynr.math)

Metrics (pynr.metrics)

Moments (pynr.moments)

Learning Rate Schedules (pynr.schedules)

Regularizers (pynr.regularizers)

Reinforcement Learning (pynr.rl)

Utilities for reinforcement learning.

Losses (pynr.rl.losses)

Targets (pynr.rl.targets)

Strategies (pynr.rl.strategies)

Wrappers (pynr.rl.wrappers)

Installation

There are a few options for installation:

  1. (Recommended) Install with pipenv:

    pipenv install fomoro-pyoneer
  2. Install locally for development with pipenv:

    git clone https://github.com/fomorians/pyoneer.git
    cd pyoneer
    pipenv install
    pipenv shell

Testing

There are a few options for testing:

  1. Run all tests:

    python -m unittest discover -bfp '*_test.py'
  2. Run specific tests:

    python -m pyoneer.math.logical_ops_test

Contributing

File an issue following the ISSUE_TEMPLATE. If the issue discussion warrants implementation, then submit a pull request from a branch describing the feature. This will eventually get merged into master after a few rounds of code review.