twitter / dict_minimize

Related projects ⓘ

Alternatives and complementary repositories for dict_minimize

• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• pyro-ppl / funsor
  Functional tensors for probabilistic programming
  ☆236Updated last year
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆58Updated 4 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆42Updated 4 years ago
• openopt / chop
  CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.
  ☆76Updated 8 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• probml / JSL
  Jax SSM Library
  ☆49Updated last year
• karalets / TyXe
  Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)
  ☆42Updated 3 years ago
• google-deepmind / einshape
  ☆99Updated 4 months ago
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆30Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 7 months ago
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆189Updated 6 months ago
• facebookresearch / pplbench
  Evaluation Framework for Probabilistic Programming Languages
  ☆98Updated 10 months ago
• google-research / ott
  ☆213Updated 2 years ago
• pierreablin / autoptim
  Automatic differentiation + optimization
  ☆104Updated 5 years ago
• LucaAmbrogioni / Brancher
  A python library for stochastic variational inference and differentiable probabilistic programming
  ☆32Updated 5 years ago
• srush / parallax
  ☆155Updated 4 years ago
• rlouf / mcx
  Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
  ☆325Updated 8 months ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• pyro-ppl / pyro-models
  Repository of models in Pyro
  ☆29Updated 3 months ago
• srush / jax-lda
  ☆30Updated 2 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 3 years ago
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆64Updated last month
• mcleonard / xaby
  Functional machine learning for fun
  ☆84Updated 3 years ago
• jax-ml / coix
  Inference Combinators in JAX
  ☆45Updated 3 weeks ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆105Updated 4 years ago
• cybertronai / autograd-lib
  ☆33Updated 4 years ago