#

jax

Here are 674 public repositories matching this topic...

niklasschmitz / DeepQLearning

small experiments with agents learning atari games, implemented in jax/numpy

deep-reinforcement-learning deep-q-learning jax

Updated Mar 16, 2019
Python

gcucurull / jax-ghnet

Graph Highway Networks in JAX

jax graph-neural-networks graph-highway-networks

Updated May 16, 2020
Python

mmcenta / jax-baselines

Implementation of core Deep Reinforcement Learning Algorithms with JAX.

python machine-learning reinforcement-learning openai gym reinforcement-learning-algorithms jax

Updated May 21, 2020
Python

jonasrauber / attax

Attax: adversarial attacks using JAX

python machine-learning adversarial-attacks jax

Updated Aug 5, 2020
Python

gante / project_euler

Python + JAX approach to the Project Euler problems

problem-solving jax

Updated Aug 29, 2020
Python

YukunXia / Carla_iLQR_MPC

Implementation of the real-time MPC based on iLQR in Carla simulator

mpc self-driving-car autonomous-vehicles trajectory-optimization carla ilqr jax

Updated Sep 3, 2020
Python

gcucurull / maml_flax

Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.

flax maml meta-learning jax

Updated Sep 18, 2020
Python

vballoli / vit-flax

Implementation of Vision Transformers in Flax

flax jax vision-transformer

Updated Oct 12, 2020
Python

juliuskunze / jaxnet

Concise deep learning for JAX

python data-science machine-learning deep-learning neural-networks jax

Updated Oct 21, 2020
Python

tristandeleu / metax

A collection of extensions for meta-learning in JAX

meta-learning jax

Updated Oct 23, 2020
Python

hanrach / p2d_solver

Solver for pseudo-2D model of Li-ion battery based on Finite difference Method. Incorporation of JAX for the nonlinear solver.

jax li-ion-battery p2d-solver

Updated Nov 19, 2020
Python

epignatelli / human-level-control-through-deep-reinforcement-learning

A jax/stax implementation of: Mnih, V., Kavukcuoglu, K., Silver, D., Rusu, A.A., Veness, J., Bellemare, M.G., Graves, A., Riedmiller, M., Fidjeland, A.K., Ostrovski, G. and Petersen, S., 2015. Human-level control through deep reinforcement learning. nature, 518(7540), pp.529-533.

reinforcement-learning deep-reinforcement-learning dqn reinforcement-learning-algorithms atari deepmind stax jax papers-with-code

Updated Dec 7, 2020
Python

branislav1991 / piper

Probabilistic programming in Python built on Google Jax

python machine-learning probabilistic-programming probabilistic-graphical-models jax

Updated Dec 19, 2020
Python

epignatelli / discovering-reinforcement-learning-algorithms

A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. and Silver, D., 2020. Discovering reinforcement learning algorithms. Advances in Neural Information Processing Systems, 33.

reinforcement-learning lstm policy-gradient rnn reinforcement-learning-algorithms stax actor-critic paper-implementations meta-learning jax paper-with-code general-meta-learning

Updated Dec 22, 2020
Python

comocheng / jax-reactor

Differentiable and hardware accelerated chemical kinetics simulations in python using JAX

automatic-differentiation jit chemical-kinetics combustion-models jax

Updated Dec 23, 2020
Python

nolanbconaway / shabadoo

Very easy Bayesian regression.

python regression bayesian-inference mcmc bayesian-statistics jax numpyro

Updated Dec 25, 2020
Python

gcucurull / jax-gcn

Graph Convolutional Networks in JAX

graph-convolutional-networks gcn jax graph-neural-networks

Updated Jan 3, 2021
Python

Numeri / jaxspp

A Spatial Pyramid Pooling library built for the jax.experimental.stax library

deep-learning convolutional-neural-networks jax spatial-pyramid-pooling

Updated Jan 11, 2021
Python

chris-tng / sasrec

Implementation of Self-Attentive Sequential Recommendation (SASRec) in flax

transformer recommender-system flax jax

Updated Jan 13, 2021
Python

maxwells-daemons / deltaflow

A GPU-accelerated differentiable fluid simulator written in JAX.

fluid-simulation jax differentiable-programming

Updated Feb 1, 2021
Python

Improve this page

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