dm-haiku
There are 21 repositories under dm-haiku topic.
kmheckel/spyx
Spyx: Spiking Neural Networks in JAX
varun-ml/diffusion-models-tutorial
Experiment with diffusion models that you can run on your local jupyter instances
evgenii-nikishin/omd
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
sail-sg/rosmo
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
deterministic-algorithms-lab/Jax-Journey
A pathway and collection of resources to learning Jax from beginning to advance.
kwk2696/sb3-jax-haiku
stable-baselines with JAX & Haiku
deterministic-algorithms-lab/NLP-Journey
This repository provides a selection of very basic and minimal notebooks for various NLP tasks in written in Jax.
ethanluoyc/td3_bc_jax
Direct port of TD3_BC to JAX using Haiku and optax.
NTT123/wavernn-16bit
The (unofficial) vanilla version of WaveRNN
Raffaelbdl/hyperbolic-nn-haiku
dm-haiku implementation of hyperbolic neural networks
epignatelli/rainbow-drl
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.
dogeplusplus/haiku-transformer
Vision Transformer implemented with JAX and dm-haiku
ethanabrooks/SAC-JAX
A JAX Implementation of the Soft Actor Critic Algorithm
hilanzy/moss
Moss is a Python library for Reinforcement Learning.
kmheckel/synecdoche
Haiku-based Hypernetworks in JAX
NTT123/pointer-networks
An unofficial implementation of pointer networks.
rystrauss/bax
A flexible trainer interface for Jax and Haiku.
Jeevesh8/AutoRegressive-MLM
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.
NTT123/haiku_trainer
A helper library for training dm-haiku models.
BirkhoffG/explainax
JAX-based Model Explanation and Interpretation Library