tristandeleu / jax-comln

Related projects: ⓘ

• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated last month
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆10Updated 3 years ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated last year
• facebookresearch / neural-scs
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Updated last year
• gcucurull / maml_flax
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆19Updated 4 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆40Updated last year
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆14Updated last year
• lili-chen / SEER
  Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.
  ☆21Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆38Updated last year
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• ThomasMiconi / Meta-Task-Generator
  Automatically generate simple meta-learning tasks from a very large space
  ☆15Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆34Updated 3 years ago
• AllanYangZhou / metalearning-symmetries
  Experiments for Meta-Learning Symmetries by Reparameterization
  ☆55Updated 3 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…
  ☆20Updated 3 years ago
• IDSIA / neuraldiffeq-fwp
  Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…
  ☆18Updated last year
• libeanim / contact-symplectic-integrator-network
  Contact-Aware Symplectic Integrator Network
  ☆11Updated last year
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated last year
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• singhgautam / snp
  This is the official source code for Sequential Neural Processes.
  ☆39Updated last year
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆26Updated 2 years ago
• sungyubkim / amortized_svgd
  A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN
  ☆18Updated 5 years ago
• sail-sg / optim4rl
  Optim4RL is a Jax framework of learning to optimize for reinforcement learning.
  ☆22Updated last month
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆31Updated 3 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year
• geyang / plan2vec
  Public Release of Plan2vec Implementation in pyTorch
  ☆56Updated last year
• twitter-research / hyperbolic-rl
  ☆56Updated last year
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆41Updated last year
• ElisevanderPol / PRAE
  Plannable Approximations to MDP Homomorphisms: Equivariance under Actions
  ☆26Updated 4 years ago
• stanfordmlgroup / disentanglement
  Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology
  ☆36Updated 3 years ago