- Notifications
You must be signed in to change notification settings - Fork0
Reinforcement Learning tooling built with Nx
License
DockYard/rein
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
⚠️ This library is a work in progress!
Reinforcement Learning algorithms written inNx.
Ifavailable in Hex, the package can be installedby addingrein
to your list of dependencies inmix.exs
:
defdepsdo[{:rein,"~> 0.1.0"}]end
This library has no external dependencies. However,one should be able to runEXLA, which is the default backend and compiler.
Torchx can also be used through theREIN_NX_BACKEND
environment variable.
REIN_NX_BACKENDIf 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.
We are very thankful for the many contributors
This library followsSemantic Versioning
At DockYard we are ready to help you build your next Elixir project. We have a unique expertisein Elixir and Phoenix development that is unmatched.Get in touch!
At DockYard we love Elixir! You canread our Elixir blog posts
DockYard, Inc. © 2023
About
Reinforcement Learning tooling built with Nx