kmheckel/spyx

Spyx: Spiking Neural Networks in JAX

Language: Jupyter Notebook - Size: 15.1 MB - Last synced at: 12 days ago - Pushed at: 11 months ago - Stars: 124 - Forks: 13

sail-sg/rosmo

Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023

Language: Python - Size: 66.4 KB - Last synced at: 2 months ago - Pushed at: about 2 years ago - Stars: 29 - Forks: 0

rystrauss/bax

A flexible trainer interface for Jax and Haiku.

Language: Python - Size: 93.8 KB - Last synced at: 3 months ago - Pushed at: over 3 years ago - Stars: 3 - Forks: 1

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.

Language: Jupyter Notebook - Size: 8.79 KB - Last synced at: 28 days ago - Pushed at: about 5 years ago - Stars: 3 - Forks: 0

BirkhoffG/explainax 📦

JAX-based Model Explanation and Interpretation Library

Language: Jupyter Notebook - Size: 439 KB - Last synced at: 2 months ago - Pushed at: over 1 year ago - Stars: 1 - Forks: 0

varun-ml/diffusion-models-tutorial

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

Language: Jupyter Notebook - Size: 259 MB - Last synced at: 10 months ago - Pushed at: 10 months ago - Stars: 54 - Forks: 8

evgenii-nikishin/omd

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

Language: Jupyter Notebook - Size: 268 KB - Last synced at: 25 days ago - Pushed at: about 4 years ago - Stars: 43 - Forks: 5

kwk2696/sb3-jax-haiku

stable-baselines with JAX & Haiku

Language: Python - Size: 297 KB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 9 - Forks: 4

ethanluoyc/td3_bc_jax

Direct port of TD3_BC to JAX using Haiku and optax.

Language: Python - Size: 5.86 KB - Last synced at: over 1 year ago - Pushed at: almost 4 years ago - Stars: 4 - Forks: 1

dogeplusplus/haiku-transformer

Vision Transformer implemented with JAX and dm-haiku

Language: Python - Size: 106 KB - Last synced at: 9 months ago - Pushed at: over 3 years ago - Stars: 2 - Forks: 0

kmheckel/synecdoche

Haiku-based Hypernetworks in JAX

Language: Python - Size: 374 KB - Last synced at: 15 days ago - Pushed at: about 2 years ago - Stars: 2 - Forks: 0

Jeevesh8/AutoRegressive-MLM Fork of deterministic-algorithms-lab/NLP-Journey

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.

Language: Python - Size: 1.91 MB - Last synced at: 7 months ago - Pushed at: over 4 years ago - Stars: 1 - Forks: 0

Raffaelbdl/hyperbolic-nn-haiku

dm-haiku implementation of hyperbolic neural networks

Language: Python - Size: 9.04 MB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 3 - Forks: 0

Raffaelbdl/minimalistic-rl

RL algorithms made simple in JAX

Language: Python - Size: 125 KB - Last synced at: over 2 years ago - Pushed at: almost 3 years ago - Stars: 4 - Forks: 0

Oaklight/jax-transformer

Language: Jupyter Notebook - Size: 1.31 MB - Last synced at: 6 months ago - Pushed at: almost 3 years ago - Stars: 0 - Forks: 0

deterministic-algorithms-lab/Jax-Journey

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

Language: Jupyter Notebook - Size: 597 KB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 11 - Forks: 0

NTT123/wavernn-16bit

The (unofficial) vanilla version of WaveRNN

Language: Python - Size: 47.9 KB - Last synced at: over 2 years ago - Pushed at: over 3 years ago - Stars: 4 - Forks: 1

NTT123/pointer-networks

An unofficial implementation of pointer networks.

Language: Jupyter Notebook - Size: 4.34 MB - Last synced at: over 2 years ago - Pushed at: almost 5 years ago - Stars: 2 - Forks: 0

ethanabrooks/SAC-JAX Fork of tesslerc/TD3-JAX

A JAX Implementation of the Soft Actor Critic Algorithm

Language: Jupyter Notebook - Size: 3.89 MB - Last synced at: over 2 years ago - Pushed at: about 5 years ago - Stars: 1 - Forks: 1

deterministic-algorithms-lab/NLP-Journey

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

Language: Jupyter Notebook - Size: 52.7 KB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 7 - Forks: 2

NTT123/haiku_trainer

A helper library for training dm-haiku models.

Language: Python - Size: 16.6 KB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 0 - Forks: 0