Search Results - RepositoryStats

12

101

mit

3

Spyx: Spiking Neural Networks in JAX

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

Created 2023-06-21

203 commits to main branch, last one 21 days ago

8

54

mit

2

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

jax vae optax pytorch dm-haiku diffusion-models

Created 2022-10-20

56 commits to master branch, last one 10 days ago

5

43

mit

3

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

dqn gym jax sac flax haiku dm-haiku deep-learning model-based-rl machine-learning soft-actor-critic reinforcement-learning automatic-differentiation deep-reinforcement-learning model-based-reinforcement-learning

Created 2021-06-06

10 commits to main branch, last one 3 years ago