stax

Here are 5 public repositories matching this topic...

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

qdevpsi3 / quantum-orthogonal-nn

Star

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

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

Star

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