Skip to content

DockYard/rein

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Rein

⚠️ This library is a work in progress!

Reinforcement Learning algorithms written in Nx.

Installation

If available in Hex, the package can be installed by adding rein to your list of dependencies in mix.exs:

def deps do
  [
    {:rein, "~> 0.1.0"}
  ]
end

Dependencies

This library has no external dependencies. However, one should be able to run EXLA, which is the default backend and compiler.

Torchx can also be used through the REIN_NX_BACKEND environment variable.

Environment variables

  • REIN_NX_BACKEND If set to "torchx", will use Torchx as the default backend. If "binary", uses plain Nx.BinaryBackend. Otherwise, will use EXLA as the default backend and compiler.

    For EXLA and Torchx, each have their own available environment variables as well.

Authors

We are very thankful for the many contributors

Versioning

This library follows Semantic Versioning

Looking for help with your Elixir project?

At DockYard we are ready to help you build your next Elixir project. We have a unique expertise in Elixir and Phoenix development that is unmatched. Get in touch!

At DockYard we love Elixir! You can read our Elixir blog posts

Legal

DockYard, Inc. © 2023

@DockYard

Licensed under the MIT license

About

Reinforcement Learning tooling built with Nx

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages