Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for reinforcement learning jax
jax
x
reinforcement-learning
x
42 search results found
D2l En
⭐
21,912
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Wandb
⭐
8,204
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Trax
⭐
7,818
Trax — Deep Learning with Clear Code and Speed
Mctx
⭐
2,201
Monte Carlo tree search in JAX
Brax
⭐
1,929
Massively parallel rigidbody physics simulation on accelerator hardware.
Dreamerv3
⭐
749
Mastering Diverse Domains through World Models
Mava
⭐
593
🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX
Jaxrl
⭐
585
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Purejaxrl
⭐
460
Really Fast End-to-End Jax RL Implementations
Jumanji
⭐
439
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
Skrl
⭐
263
Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Isaac Orbit and Omniverse Isaac Gym
Pgx
⭐
260
🎲 Vectorized RL game environments in JAX [NeurIPS23]
Qdax
⭐
233
Accelerated Quality-Diversity
Indaba Pracs 2022
⭐
151
Notebooks for the Practicals at the Deep Learning Indaba 2022.
Xland Minigrid
⭐
104
JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
Magi
⭐
103
Reinforcement learning library in JAX.
Flashbax
⭐
100
⚡ Flashbax: Accelerated Replay Buffers in JAX
Adam
⭐
82
adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
Reinforced Lib
⭐
60
Reinforcement learning library
Turbozero
⭐
53
fast + parallel AlphaZero in JAX
Jax Rl
⭐
52
JAX implementations of core Deep RL algorithms
Wax Ml
⭐
38
A Python library for machine-learning and feedback loops on streaming data
Sac N Jax
⭐
36
Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
Redco
⭐
35
MLSys Workshop NeurIPS 2023 - Redco: A Lightweight Tool to Automate Distributed Training and Inference
Fastpbrl
⭐
31
Vectorization techniques for fast population-based training.
Sebulba
⭐
31
🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
Pyhgf
⭐
27
PyHGF: A graph neural network library for predictive coding
Corax
⭐
25
Corax: Core RL in JAX
Omd
⭐
23
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Xpag
⭐
20
a modular reinforcement learning library with JAX agents
Jaxcql
⭐
20
Conservative Q learning in Jax
Fisor
⭐
19
[ICLR 2024] The official implementation of "Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Model"
Rosmo
⭐
19
Code for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
Omega
⭐
16
A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.
Popjaxrl
⭐
16
Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
Td3 Jax
⭐
11
A JAX Implementation of the Twin Delayed DDPG Algorithm
Rl_with_resets
⭐
11
JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
Helx
⭐
9
Interoperating between (Deep) Reiforcement Learning libraries
Dopamax
⭐
9
Reinforcement learning in pure JAX.
Rl Basics
⭐
8
Simple single file implementation of Deep Reinforcement Learning algorithms
Animus
⭐
7
Minimalistic framework to run machine learning experiments.
Sb3 Jax Haiku
⭐
7
stable-baselines with JAX & Haiku
Related Searches
Python Reinforcement Learning (2,612)
Jupyter Notebook Reinforcement Learning (761)
Pytorch Reinforcement Learning (681)
Reinforcement Learning Rl (649)
Artificial Intelligence Reinforcement Learning (623)
Reinforcement Learning Gym (508)
Reinforcement Learning Openai Gym (378)
Video Game Reinforcement Learning (326)
1-42 of 42 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.