dm-haiku

There are 0 repository under dm-haiku topic.

spyx
kmheckel / spyx
Spyx: Spiking Neural Networks in JAX
ai dm-haiku jax ml neural-networks neuromorphic snn spiking-neural-networks
Language:Jupyter Notebook 80
diffusion-models-tutorial
varun-ml / diffusion-models-tutorial
Experiment with diffusion models that you can run on your local jupyter instances
diffusion-models dm-haiku jax optax pytorch vae
Language:Jupyter Notebook 49
evgenii-nikishin / omd
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
reinforcement-learning deep-learning deep-reinforcement-learning gym flax jax dm-haiku haiku machine-learning sac soft-actor-critic model-based-rl model-based-reinforcement-learning dqn automatic-differentiation
Language:Jupyter Notebook 44
sail-sg / rosmo
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
atari model-based-rl muzero offline-reinforcement-learning reinforcement-learning rl-unplugged jax dm-haiku arcade-learning-environment bsuite model-based-reinforcement-learning muzero-unplugged offline-rl
Language:Python 26
deterministic-algorithms-lab / Jax-Journey
A pathway and collection of resources to learning Jax from beginning to advance.
deep-learning dm-haiku flax jax machine-learning
Language:Jupyter Notebook 11
kwk2696 / sb3-jax-haiku
stable-baselines with JAX & Haiku
dm-haiku haiku imitation-learning jax reinforcement-learning decision-transformers soft-actor-critic proximal-policy-optimization behavior-cloning dataset-aggregation diffusion
Language:Python 9
deterministic-algorithms-lab / NLP-Journey
This repository provides a selection of very basic and minimal notebooks for various NLP tasks in written in Jax.
jax dm-haiku nlp transformers language-modeling classification nmt
Language:Jupyter Notebook 7
ethanluoyc / td3_bc_jax
Direct port of TD3_BC to JAX using Haiku and optax.
offline-rl td3 jax dm-haiku optax reinforcement-learning
Language:Python 4
NTT123 / wavernn-16bit
The (unofficial) vanilla version of WaveRNN
wavernn vocoder speech-synthesis neural-network gru rnn dm-haiku jax optax
Language:Python 4
Raffaelbdl / hyperbolic-nn-haiku
dm-haiku implementation of hyperbolic neural networks
dm-haiku hyperbolic-geometry jax neural-networks optax optimizers
Language:Python 4
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.
deep-learning deep-reinforcement-learning dm-haiku dqn dqn-jax haiku jax papers-with-code reinforcement-learning
Language:Jupyter Notebook 3
dogeplusplus / haiku-transformer
Vision Transformer implemented with JAX and dm-haiku
computer-vision dm-haiku jax transformers
Language:Python 2
ethanabrooks / SAC-JAX
A JAX Implementation of the Soft Actor Critic Algorithm
deep-reinforcement-learning dm-haiku haiku jax reinforcement-learning soft-actor-critic
Language:Jupyter Notebook 2
hilanzy / moss
Moss is a Python library for Reinforcement Learning.
atari envpool impala jax reinforcement-learning rlax dm-haiku dm-launchpad library pypi pettingzoo marl ppo
Language:Python 2
kmheckel / synecdoche
Haiku-based Hypernetworks in JAX
dm-haiku hypernetworks jax
Language:Python 2
NTT123 / pointer-networks
An unofficial implementation of pointer networks.
convex-hull deep-learning dm-haiku haiku-trainer jax pca pointer-networks ptr-net
Language:Jupyter Notebook 2
rystrauss / bax
A flexible trainer interface for Jax and Haiku.
boilerplate dm-haiku jax trainer
Language:Python 2
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.
mlm argumentation nlp deep-learning jax dm-haiku language-modeling
Language:Python 1
NTT123 / haiku_trainer
A helper library for training dm-haiku models.
dm-haiku deep-learning trainer jax optax machine-learning haiku-trainer
Language:Python 1
BirkhoffG / explainax
JAX-based Model Explanation and Interpretation Library
explainability explainable-ai explainable-ml interpretability interpretable-machine-learning jax machine-learning flax deep-learning dm-haiku interpretable-ai interpretable-ml python
Language:Jupyter Notebook 0
Oaklight / jax-transformer
attention-is-all-you-need dm-haiku jax transformer
Language:Jupyter Notebook 0