3 results found Sort:

kmheckel
spyx kmheckel
12
102
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 about a month ago
varun-ml
diffusion-models-tutorial varun-ml
8
55
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 26 days ago
evgenii-nikishin
omd evgenii-nikishin
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