Alternatives and similar repositories for jax-comln

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

• gcucurull / maml_flax

  View on GitHub
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆21Sep 18, 2020Updated 5 years ago
• alexub / jax-meta-learning

  View on GitHub
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆11Oct 6, 2020Updated 5 years ago
• evgenii-nikishin / omd

  View on GitHub
  JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
  ☆44Jun 14, 2021Updated 4 years ago
• BY571 / OFENet

  View on GitHub
  ☆10Mar 22, 2021Updated 4 years ago
• danieljtait / ladax

  View on GitHub
  Layered distributions using FLAX/JAX
  ☆10Dec 13, 2020Updated 5 years ago
• DiffEqML / tutorials

  View on GitHub
  ☆11Apr 14, 2022Updated 3 years ago
• mlysy / rodeo-legacy

  View on GitHub
  Probabilistic Solution of Differential Equations
  ☆13Jun 19, 2022Updated 3 years ago
• jimimvp / torch_rl

  View on GitHub
  Reinforcement learning library for PyTorch.
  ☆11Jun 15, 2018Updated 7 years ago
• rythei / DARLA-PyTorch

  View on GitHub
  PyTorch implementation of DARLA preprocessing models
  ☆11Jan 30, 2018Updated 8 years ago
• schroederdewitt / mackrl

  View on GitHub
  Code for Multi-Agent Common Knowledge Reinforcement Learning (NeurIPS 2019)
  ☆33Dec 1, 2019Updated 6 years ago
• rgoldade / GeometricMultigridPressureSolver

  View on GitHub
  ☆14Sep 7, 2020Updated 5 years ago
• ravi-lanka-4 / CoPiEr

  View on GitHub
  Co-training for Policy Learning
  ☆13Aug 8, 2019Updated 6 years ago
• bhairavmehta95 / ant-env

  View on GitHub
  Ant Gather and Ant Maze envs, separated from RLLab
  ☆11Aug 2, 2018Updated 7 years ago
• kazukiosawa / ngd_in_wide_nn

  View on GitHub
  simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…
  ☆15Oct 21, 2020Updated 5 years ago
• myagues / flax_nerf

  View on GitHub
  Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API
  ☆13Sep 25, 2021Updated 4 years ago
• jinxu06 / gsubsampling

  View on GitHub
  Reference implementation for "Group Equivariant Subsampling"
  ☆16Dec 13, 2021Updated 4 years ago
• aparis69 / Implicit-Volumetric-Terrains

  View on GitHub
  Source code for the TOG paper: Terrain Amplification with Implicit 3D Features. Presented at Siggraph Asia 2019.
  ☆17May 20, 2022Updated 3 years ago
• philipjball / ReadyPolicyOne

  View on GitHub
  🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)
  ☆18Jul 6, 2023Updated 2 years ago
• aterenin / SparseGPAX

  View on GitHub
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Mar 31, 2021Updated 4 years ago
• largelymfs / svpg_REINFORCE

  View on GitHub
  Stein Variational Policy Gradient for REINFORCE
  ☆18Jul 12, 2017Updated 8 years ago
• BerkeleyAutomation / DART

  View on GitHub
  ☆50Nov 26, 2019Updated 6 years ago
• ericjang / maml-jax

  View on GitHub
  Implementation of Model-Agnostic Meta-Learning (MAML) in Jax
  ☆191Sep 28, 2022Updated 3 years ago
• ethanluoyc / e2c-pytorch

  View on GitHub
  E2C implementation in PyTorch
  ☆43Jul 5, 2017Updated 8 years ago
• qsh-zh / pis

  View on GitHub
  [ICLR 2022] Path integral sampler
  ☆52Aug 21, 2023Updated 2 years ago
• tristandeleu / jax-meta-learning

  View on GitHub
  A collection of meta-learning algorithms in Jax
  ☆24Sep 3, 2022Updated 3 years ago
• PrincetonLIPS / AHGP

  View on GitHub
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆23Dec 9, 2020Updated 5 years ago
• KeunwooPark / siren-jax

  View on GitHub
  Unofficial implementation of Siren with Jax for image representation.
  ☆19Dec 10, 2021Updated 4 years ago
• epignatelli / discovering-reinforcement-learning-algorithms

  View on GitHub
  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…
  ☆23Dec 22, 2020Updated 5 years ago
• IvanYashchuk / jax-fenics-adjoint

  View on GitHub
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆107Nov 12, 2023Updated 2 years ago
• FxnlBiomechLab / twsa-optimization-for-rra

  View on GitHub
  An evolutionary algorithm-based optimization for tracking weights in the OpenSim Residual Reduction Algorithm (RRA).
  ☆11Jul 17, 2023Updated 2 years ago
• tonyzhaozh / meld

  View on GitHub
  MELD: Meta-Reinforcement Learning from Images via Latent State Models https://arxiv.org/abs/2010.13957
  ☆67Apr 30, 2021Updated 4 years ago
• greydanus / optimize_wing

  View on GitHub
  We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.
  ☆27Oct 16, 2020Updated 5 years ago
• gyh75520 / Relational_DRL

  View on GitHub
  Implementation of Relational Deep Reinforcement Learning
  ☆25Jan 31, 2020Updated 6 years ago
• cambridge-mlg / convcnp

  View on GitHub
  Implementation of the Convolutional Conditional Neural Process
  ☆128May 17, 2021Updated 4 years ago
• VincentYu68 / policy_transfer

  View on GitHub
  ☆30Jun 10, 2020Updated 5 years ago
• jeappen / gym-grid

  View on GitHub
  A simple Gridworld environment for Open AI gym
  ☆25Jun 10, 2018Updated 7 years ago
• ermongroup / fast_feedforward_computation

  View on GitHub
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆29Sep 25, 2021Updated 4 years ago
• Wenxuan-Zhou / EPI

  View on GitHub
  Code for Environment Probing Interaction Policies [ICLR 2019]
  ☆29Jun 17, 2019Updated 6 years ago
• srush / jax-lda

  View on GitHub
  ☆30Feb 11, 2022Updated 4 years ago