#

stax

Here are 5 public repositories matching this topic...

epignatelli / human-level-control-through-deep-reinforcement-learning

A jax/stax implementation of: Mnih, V., Kavukcuoglu, K., Silver, D., Rusu, A.A., Veness, J., Bellemare, M.G., Graves, A., Riedmiller, M., Fidjeland, A.K., Ostrovski, G. and Petersen, S., 2015. Human-level control through deep reinforcement learning. nature, 518(7540), pp.529-533.

reinforcement-learning deep-reinforcement-learning dqn reinforcement-learning-algorithms atari deepmind stax jax papers-with-code

Updated Dec 7, 2020
Python

epignatelli / discovering-reinforcement-learning-algorithms

A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. and Silver, D., 2020. Discovering reinforcement learning algorithms. Advances in Neural Information Processing Systems, 33.

reinforcement-learning lstm policy-gradient rnn reinforcement-learning-algorithms stax actor-critic paper-implementations meta-learning jax paper-with-code general-meta-learning

Updated Dec 22, 2020
Python

jifffffy / STAX

STAF eXecute engine with pf4j plugin framework to support easy develop plugins

plugin xml engine xfs stax staf

Updated Jun 30, 2021
Python

qdevpsi3 / quantum-orthogonal-nn

JAX implementation of Classical and Quantum Algorithms for Orthogonal Neural Networks by (Kerenidis et al., 2021)

quantum-computing mnist-classification haiku stax jax optax quantum-inspired orthogonal-neural-network

Updated Jan 6, 2023
Python

stax-labs / lib-stax-python-sdk

aws stax sdk-python staxapp

Updated Sep 9, 2024
Python

Improve this page

Add a description, image, and links to the stax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the stax topic, visit your repo's landing page and select "manage topics."