#

dm-haiku

Here are 10 public repositories matching this topic...

sail-sg / rosmo

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

kwk2696 / sb3-jax-haiku

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

ethanluoyc / td3_bc_jax

Direct port of TD3_BC to JAX using Haiku and optax.

reinforcement-learning jax td3 dm-haiku offline-rl optax

Updated Sep 7, 2021
Python

Raffaelbdl / hyperbolic-nn-haiku

dm-haiku implementation of hyperbolic neural networks

neural-networks hyperbolic-geometry jax optimizers dm-haiku optax

Updated Jan 29, 2023
Python

NTT123 / wavernn-16bit

The (unofficial) vanilla version of WaveRNN

neural-network speech-synthesis gru rnn vocoder jax wavernn dm-haiku optax

Updated Dec 7, 2021
Python

kmheckel / synecdoche

Haiku-based Hypernetworks in JAX

hypernetworks jax dm-haiku

Updated Jun 25, 2023
Python

rystrauss / bax

A flexible trainer interface for Jax and Haiku.

boilerplate trainer jax dm-haiku

Updated May 30, 2022
Python

dogeplusplus / haiku-transformer

Vision Transformer implemented with JAX and dm-haiku

computer-vision transformers jax dm-haiku

Updated Mar 31, 2022
Python

NTT123 / haiku_trainer

A helper library for training dm-haiku models.

machine-learning deep-learning trainer jax dm-haiku optax haiku-trainer

Updated Feb 7, 2021
Python

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.

nlp deep-learning language-modeling mlm argumentation jax dm-haiku

Updated May 23, 2021
Python

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."