Alternatives and similar repositories for jaxde

Users that are interested in jaxde are comparing it to the libraries listed below

• adamhaber / JaxEnt

  View on GitHub
  Jax-based MaxEnt
  ☆17Nov 24, 2019Updated 6 years ago
• joschu / jax-exp

  View on GitHub
  ☆24Feb 3, 2019Updated 7 years ago
• PrincetonLIPS / randomized_telescopes

  View on GitHub
  Code for "Efficient optimization of loops and limits with randomized telescoping sums"
  ☆27May 13, 2019Updated 6 years ago
• danieljtait / jax_xla_adventures

  View on GitHub
  Dive into Jax, Flax, XLA and C++
  ☆32Apr 1, 2020Updated 5 years ago
• pomonam / NoisyNaturalGradient

  View on GitHub
  TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".
  ☆60Jan 12, 2019Updated 7 years ago
• hanxzh94 / matrix-model

  View on GitHub
  Variational Monte Carlo with generative flows for mini-BMN matrix models
  ☆19Aug 20, 2020Updated 5 years ago
• ericjang / maml-jax

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

  View on GitHub
  A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions
  ☆22May 5, 2019Updated 6 years ago
• juliagusak / neural-ode-metasolver

  View on GitHub
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Mar 30, 2021Updated 4 years ago
• vlfeat / researchdoom-chocolate

  View on GitHub
  ResearchDoom fork of the Chocolate Doom engine.
  ☆16Oct 20, 2017Updated 8 years ago
• mattjj / ode-diff-notes

  View on GitHub
  ☆12Oct 8, 2020Updated 5 years ago
• csc2541-f17 / csc2541-f17.github.io

  View on GitHub
  ☆12Dec 7, 2017Updated 8 years ago
• frankwswang / QuMGAN.jl

  View on GitHub
  Quantum GAN algorithm based on MPS, realized by Julia.
  ☆11Apr 5, 2019Updated 6 years ago
• ericjang / nf-jax

  View on GitHub
  Normalizing Flows in Jax
  ☆107Aug 19, 2020Updated 5 years ago
• Joshuaalbert / jaxnlds

  View on GitHub
  Inference on non-linear dynamical systems written in JAX
  ☆11Aug 20, 2020Updated 5 years ago
• gcucurull / jax-gat

  View on GitHub
  JAX implementation of Graph Attention Networks
  ☆13Jan 29, 2022Updated 4 years ago
• conormdurkan / autoregressive-energy-machines

  View on GitHub
  Autoregressive Energy Machines
  ☆78Nov 22, 2022Updated 3 years ago
• caiyunapp / leibniz

  View on GitHub
  Leibniz is a python package which provide facilities to express learnable partial differential equations with PyTorch
  ☆15Sep 13, 2021Updated 4 years ago
• rgiordan / paragami

  View on GitHub
  "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.
  ☆14Jan 23, 2024Updated 2 years ago
• jiamings / ais

  View on GitHub
  Annealed Importance Sampling (AIS) for generative models.
  ☆16Jul 20, 2018Updated 7 years ago
• IvanYashchuk / jax-firedrake

  View on GitHub
  Differentiable interface to Firedrake for JAX
  ☆15Feb 28, 2021Updated 4 years ago
• andymiller / ReducedVarianceReparamGradients

  View on GitHub
  Reducing Reparameterization Gradient Variance code.
  ☆33Jun 5, 2017Updated 8 years ago
• asteroidhouse / self-tuning-networks

  View on GitHub
  Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088
  ☆61Jun 18, 2019Updated 6 years ago
• Zhouquan-Wan / SVD_autodiff

  View on GitHub
  ☆18Nov 25, 2021Updated 4 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
• chaserileyroberts / TensorNetsForML

  View on GitHub
  Code for the paper "Tensor Networks for Maching Learning"
  ☆17Nov 7, 2019Updated 6 years ago
• slimgroup / Software.SEG2020

  View on GitHub
  Software for SEG2020 abstracts
  ☆15Apr 19, 2021Updated 4 years ago
• juliagusak / neural-ode-norm

  View on GitHub
  Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"
  ☆16Apr 27, 2020Updated 5 years ago
• ofnote / tsanley

  View on GitHub
  A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
  ☆40Nov 9, 2019Updated 6 years ago
• Bonnevie / rebar

  View on GitHub
  A Python implementation of the gradient REBAR estimator.
  ☆47Jun 13, 2018Updated 7 years ago
• alecwangcq / EigenDamage-Pytorch

  View on GitHub
  Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934
  ☆113Mar 3, 2020Updated 5 years ago
• dilinwang820 / adaptive-f-divergence

  View on GitHub
  A tensorflow implementation of the NIPS 2018 paper "Variational Inference with Tail-adaptive f-Divergence"
  ☆20Jan 11, 2019Updated 7 years ago
• tonyduan / transformer-blocks

  View on GitHub
  Multi-Head Attention, Transformer, Perceiver, Linear Attention.
  ☆12Oct 24, 2023Updated 2 years ago
• KyriacosShiarli / taco

  View on GitHub
  ☆25Jan 2, 2019Updated 7 years ago
• FluxML / YaoFlux.jl

  View on GitHub
  Differentiable programming on quantum circuits with Flux
  ☆19Feb 12, 2020Updated 6 years ago
• wellecks / vaes

  View on GitHub
  Variational Autoencoders & Normalizing Flows Project
  ☆18Dec 16, 2016Updated 9 years ago
• lezcano / expRNN

  View on GitHub
  Optimization with orthogonal constraints and on general manifolds
  ☆130Jul 13, 2020Updated 5 years ago
• johannbrehmer / higgs_inference

  View on GitHub
  Code repository for the paper "Constraining Effective Field Theories with Machine Learning"
  ☆21Sep 11, 2019Updated 6 years ago
• refraction-ray / admc

  View on GitHub
  Infinite order automatic differentiation for Monte Carlo with unnormalized probability distribution
  ☆24Mar 24, 2023Updated 2 years ago