justinjfu / jaxutil

Related projects: ⓘ

• gehring / fax
  ☆78Updated 3 years ago
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 4 years ago
• danieljtait / ladax
  Layered distributions using FLAX/JAX
  ☆10Updated 3 years ago
• tbrx / compiled-inference
  Train neural networks to use as SMC and importance sampling proposals
  ☆24Updated 6 years ago
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆57Updated 4 years ago
• PrincetonLIPS / randomized_telescopes
  Code for "Efficient optimization of loops and limits with randomized telescoping sums"
  ☆27Updated 5 years ago
• pawni / gpuobserver
  ☆29Updated 3 years ago
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆12Updated 4 years ago
• conormdurkan / autoregressive-energy-machines
  Autoregressive Energy Machines
  ☆77Updated last year
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆40Updated 4 years ago
• PrincetonLIPS / RandomizedAutomaticDifferentiation
  Experiment code for "Randomized Automatic Differentiation"
  ☆67Updated 4 years ago
• adamhaber / JaxEnt
  Jax-based MaxEnt
  ☆18Updated 4 years ago
• Kajiyu / kanerva_machine
  The implementation of "The Kanerva Machine" with Pytorch and Pyro
  ☆12Updated 6 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆29Updated 4 years ago
• mfinzi / residual-pathway-priors
  ☆17Updated 2 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆34Updated 3 years ago
• slinderman / stats215
  ☆29Updated 4 years ago
• lxuechen / BDMC
  PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.
  ☆52Updated 3 years ago
• PrincetonLIPS / reversible-inductive-construction
  Discrete Object Generation with Reversible Inductive Construction (NeurIPS 2019)
  ☆30Updated 3 years ago
• microsoft / statopt
  Statistical adaptive stochastic optimization methods
  ☆32Updated 4 years ago
• j-towns / fastar
  ☆15Updated 3 years ago
• locuslab / monotone_op_net
  Monotone operator equilibrium networks
  ☆51Updated 4 years ago
• franrruiz / uivi
  Code for Unbiased Implicit Variational Inference (UIVI)
  ☆13Updated 5 years ago
• pomonam / NoisyNaturalGradient
  TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".
  ☆60Updated 5 years ago
• rgiordan / paragami
  "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.
  ☆13Updated 7 months ago
• cunningham-lab / cb_and_cc
  ☆35Updated last year
• AllanYangZhou / metalearning-symmetries
  Experiments for Meta-Learning Symmetries by Reparameterization
  ☆55Updated 3 years ago
• lxuechen / inference-suboptimality
  Code for "Inference Suboptimality in Variational Autoencoders"
  ☆14Updated 4 years ago
• plai-group / tvo
  Code for the Thermodynamic Variational Objective
  ☆26Updated 2 years ago
• MethodsOfMachineLearning / probabilistic_line_search
  Probabilistic line search algorithm for stochastic optimization with a TensorFlow interface.
  ☆22Updated 7 years ago