Here are
20 public repositories
matching this topic...
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.
Updated
Jul 20, 2020
Jupyter Notebook
A JAX Implementation of the Soft Actor Critic Algorithm
Updated
Aug 19, 2020
Jupyter Notebook
An unofficial implementation of pointer networks.
Updated
Dec 6, 2020
Jupyter Notebook
A pathway and collection of resources to learning Jax from beginning to advance.
Updated
Jan 2, 2021
Jupyter Notebook
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
A helper library for training dm-haiku models.
Updated
Feb 7, 2021
Python
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
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Updated
Jun 14, 2021
Jupyter Notebook
Direct port of TD3_BC to JAX using Haiku and optax.
Updated
Sep 7, 2021
Python
The (unofficial) vanilla version of WaveRNN
Updated
Dec 7, 2021
Python
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
Updated
Oct 7, 2022
Jupyter Notebook
dm-haiku implementation of hyperbolic neural networks
Updated
Jan 29, 2023
Python
Haiku-based Hypernetworks in JAX
Updated
Jun 25, 2023
Python
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
Updated
Jul 18, 2023
Python
JAX-based Model Explanation and Interpretation Library
Updated
Apr 10, 2024
Jupyter Notebook
stable-baselines with JAX & Haiku
Updated
Jun 20, 2024
Python
Experiment with diffusion models that you can run on your local jupyter instances
Updated
Aug 8, 2024
Jupyter Notebook
Spyx: Spiking Neural Networks in JAX
Updated
Aug 16, 2024
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.
Curate this topic
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."
Learn more
You can’t perform that action at this time.