Alternatives and similar repositories for jax-comln:

Users that are interested in jax-comln are comparing it to the libraries listed below

• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated 7 months ago
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆9Updated 4 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• lili-chen / SEER
  Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.
  ☆21Updated 4 years ago
• facebookresearch / neural-scs
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Updated 2 years ago
• libeanim / contact-symplectic-integrator-network
  Contact-Aware Symplectic Integrator Network
  ☆12Updated 2 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated last year
• gcucurull / maml_flax
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆19Updated 4 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 4 months ago
• geyang / plan2vec
  Public Release of Plan2vec Implementation in pyTorch
  ☆56Updated 2 years ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 2 years ago
• singhgautam / snp
  This is the official source code for Sequential Neural Processes.
  ☆40Updated 2 years ago
• sungyubkim / amortized_svgd
  A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN
  ☆18Updated 6 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆13Updated 3 years ago
• philipjball / ReadyPolicyOne
  🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)
  ☆18Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• 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. a…
  ☆21Updated 4 years ago
• franrruiz / uivi
  Code for Unbiased Implicit Variational Inference (UIVI)
  ☆14Updated 6 years ago
• IDSIA / neuraldiffeq-fwp
  Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…
  ☆20Updated 2 years ago
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated last year
• distillpub / post--understanding-rl-vision
  Understanding RL vision Distill article
  ☆23Updated 2 years ago
• evgenii-nikishin / omd
  JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
  ☆43Updated 3 years ago
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆28Updated 2 years ago
• two2tee / WorldModelPlanning
  ☆16Updated 4 years ago
• a7b23 / H-Divergence
  ☆21Updated 2 years ago
• joeybose / FloRL
  Implicit Normalizing Flows + Reinforcement Learning
  ☆60Updated 5 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆45Updated last year
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago