Movatterモバイル変換

Skip to content

#

dm-haiku

Here are 20 public repositories matching this topic...

Language:All

Filter by language

All20 Jupyter Notebook10 Python10

Sort:Most stars

Sort options

Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

spyx

kmheckel /spyx

Spyx: Spiking Neural Networks in JAX

ai ml neural-networks spiking-neural-networks neuromorphic snn jax dm-haiku

UpdatedOct 16, 2024
Jupyter Notebook

diffusion-models-tutorial

varun-ml /diffusion-models-tutorial

Experiment with diffusion models that you can run on your local jupyter instances

pytorch vae jax diffusion-models dm-haiku optax

UpdatedOct 27, 2024
Jupyter Notebook

evgenii-nikishin /omd

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

UpdatedJun 14, 2021
Jupyter Notebook

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

UpdatedJul 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

UpdatedJun 20, 2024
Python

deterministic-algorithms-lab /Jax-Journey

A pathway and collection of resources to learning Jax from beginning to advance.

machine-learning deep-learning flax jax dm-haiku

UpdatedJan 2, 2021
Jupyter Notebook

deterministic-algorithms-lab /NLP-Journey

This repository provides a selection of very basic and minimal notebooks for various NLP tasks in written in Jax.

nlp transformers language-modeling classification nmt jax dm-haiku

UpdatedJan 14, 2021
Jupyter Notebook

Raffaelbdl /hyperbolic-nn-haiku

dm-haiku implementation of hyperbolic neural networks

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

UpdatedJan 29, 2023
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

UpdatedSep 7, 2021
Python

NTT123 /wavernn-16bit

The (unofficial) vanilla version of WaveRNN

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

UpdatedDec 7, 2021
Python

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.

reinforcement-learning deep-learning deep-reinforcement-learning dqn haiku jax papers-with-code dm-haiku dqn-jax

UpdatedJul 20, 2020
Jupyter Notebook

rystrauss /bax

A flexible trainer interface for Jax and Haiku.

boilerplate trainer jax dm-haiku

UpdatedMay 30, 2022
Python

NTT123 /pointer-networks

An unofficial implementation of pointer networks.

deep-learning pointer-networks pca convex-hull jax ptr-net dm-haiku haiku-trainer

UpdatedDec 6, 2020
Jupyter Notebook

kmheckel /synecdoche

Haiku-based Hypernetworks in JAX

hypernetworks jax dm-haiku

UpdatedJun 25, 2023
Python

dogeplusplus /haiku-transformer

Vision Transformer implemented with JAX and dm-haiku

computer-vision transformers jax dm-haiku

UpdatedMar 31, 2022
Python

ethanabrooks /SAC-JAX

A JAX Implementation of the Soft Actor Critic Algorithm

reinforcement-learning deep-reinforcement-learning haiku jax soft-actor-critic dm-haiku

UpdatedAug 19, 2020
Jupyter Notebook

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

UpdatedMay 23, 2021
Python

NTT123 /haiku_trainer

A helper library for training dm-haiku models.

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

UpdatedFeb 7, 2021
Python

BirkhoffG /explainax

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

UpdatedApr 10, 2024
Jupyter Notebook

Oaklight /jax-transformer

transformer attention-is-all-you-need jax dm-haiku

UpdatedOct 7, 2022
Jupyter Notebook

Improve this page

Add a description, image, and links to thedm-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 thedm-haiku topic, visit your repo's landing page and select "manage topics."

[8]ページ先頭

©2009-2025 Movatter.jp