Spyx: Spiking Neural Networks in JAX
-
Updated
Oct 16, 2024 - Jupyter Notebook
#
dm-haiku
Here are 20 public repositories matching this topic...
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
machine-learning reinforcement-learning deep-learning automatic-differentiation deep-reinforcement-learning dqn gym flax haiku sac model-based-rl jax model-based-reinforcement-learning soft-actor-critic dm-haiku
-
Updated
Jun 14, 2021 - Jupyter Notebook
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
reinforcement-learning atari arcade-learning-environment model-based-rl jax model-based-reinforcement-learning bsuite muzero dm-haiku offline-rl offline-reinforcement-learning rl-unplugged muzero-unplugged
-
Updated
Jul 18, 2023 - Python
A pathway and collection of resources to learning Jax from beginning to advance.
-
Updated
Jan 2, 2021 - Jupyter Notebook
stable-baselines with JAX & Haiku
reinforcement-learning imitation-learning diffusion haiku dataset-aggregation proximal-policy-optimization behavior-cloning jax soft-actor-critic dm-haiku decision-transformers
-
Updated
Jun 20, 2024 - Python
This repository provides a selection of very basic and minimal notebooks for various NLP tasks in written in Jax.
-
Updated
Jan 14, 2021 - Jupyter Notebook
Direct port of TD3_BC to JAX using Haiku and optax.
-
Updated
Sep 7, 2021 - Python
dm-haiku implementation of hyperbolic neural networks
-
Updated
Jan 29, 2023 - Python
This is a python JAX implementation of the paper: Rainbow: Combining improvements in deep reinforcement learning, by M. Hessel et al. In Thirty-Second AAAI Conference on Artificial Intelligence.
reinforcement-learning deep-learning deep-reinforcement-learning dqn haiku jax papers-with-code dm-haiku dqn-jax
-
Updated
Jul 20, 2020 - Jupyter Notebook
The (unofficial) vanilla version of WaveRNN
-
Updated
Dec 7, 2021 - Python
An unofficial implementation of pointer networks.
-
Updated
Dec 6, 2020 - Jupyter Notebook
Vision Transformer implemented with JAX and dm-haiku
-
Updated
Mar 31, 2022 - Python
A flexible trainer interface for Jax and Haiku.
-
Updated
May 30, 2022 - Python
A JAX Implementation of the Soft Actor Critic Algorithm
-
Updated
Aug 19, 2020 - Jupyter Notebook
This repository extends a basic MLM implementation to allow for efficiently conditioning on chained previous texts, in a tree; for e.g., a Reddit thread.
-
Updated
May 23, 2021 - Python
A helper library for training dm-haiku models.
-
Updated
Feb 7, 2021 - Python
JAX-based Model Explanation and Interpretation Library
python machine-learning deep-learning flax interpretability interpretable-ai interpretable-ml explainable-ai explainable-ml jax interpretable-machine-learning explainability dm-haiku
-
Updated
Apr 10, 2024 - Jupyter Notebook
-
Updated
Oct 7, 2022 - Jupyter Notebook
Improve this page
Add a description, image, and links to the dm-haiku topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the dm-haiku topic, visit your repo's landing page and select "manage topics."