Alternatives and similar repositories for jax-lda:

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

• mblondel / projection-losses
  Python implementation of projection losses.
  ☆26Updated 5 years ago
• harvardnlp / strux
  ☆18Updated 3 years ago
• timvieira / rl
  Reference implementation of algorithms for reinforcement learning and Markov decision processes.
  ☆12Updated 4 years ago
• srush / learns-dex
  ☆33Updated 4 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 5 years ago
• anirudh9119 / walkback_nips17
  Variational Walkback, NIPS'17
  ☆28Updated 7 years ago
• PrincetonLIPS / randomized_telescopes
  Code for "Efficient optimization of loops and limits with randomized telescoping sums"
  ☆27Updated 5 years ago
• plai-group / covid
  Modelling epidemiological dynamics and performing inference in these models
  ☆27Updated 3 years ago
• matejbalog / gumbel-relatives
  Code to reproduce experiments appearing in the academic paper Lost Relatives of the Gumbel Trick
  ☆17Updated 7 years ago
• adjidieng / REM
  Reweighted Expectation Maximization
  ☆29Updated 5 years ago
• google-research / autoconj
  Recognizing and exploiting conjugacy without a domain-specific language
  ☆36Updated 5 years ago
• RemiLeblond / SeaRNN-open
  Open source implementation of SeaRNN (ICLR 2018, https://openreview.net/forum?id=HkUR_y-RZ)
  ☆48Updated 6 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• harvardnlp / hmm-lm
  ☆42Updated 4 years ago
• cmaddis / astar-sampling
  A generic Monte Carlo method based on the Gumbel-Max trick.
  ☆32Updated 8 years ago
• kyunghyuncho / backprop-kalman-filter
  ☆45Updated 5 years ago
• nikitakit / sabertooth
  Standalone pre-training recipe with JAX+Flax
  ☆31Updated 2 years ago
• deep-spin / lp-sparsemap
  LP-SparseMAP: Differentiable sparse structured prediction in coarse factor graphs
  ☆41Updated last year
• ofnote / tsanley
  A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
  ☆40Updated 5 years ago
• pyro-ppl / pyro-models
  Repository of models in Pyro
  ☆29Updated 9 months ago
• kyunghyuncho / jax-practice
  ☆12Updated 4 years ago
• cgraywang / transformer-on-diet
  Code repo for "Transformer on a Diet" paper
  ☆31Updated 4 years ago
• KarenUllrich / binary-VAE
  A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.
  ☆22Updated 5 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• stanis-morozov / prodige
  A supplementary code for Beyond Vector Spaces: Compact Data Representation as Differentiable Weighted Graphs.
  ☆47Updated 5 years ago
• jacobandreas / rnn-syn
  Analogs of Linguistic Structure in Deep Representations
  ☆19Updated 7 years ago
• namedtensor / notation
  ☆108Updated 2 years ago
• rotinov / AGNES
  Flexible Reinforcement Learning Framework with PyTorch
  ☆22Updated 4 years ago
• rwightman / py-lmdb
  Universal Python binding for the LMDB 'Lightning' Database
  ☆13Updated 7 years ago
• srush / torch-golf
  Silly twitter torch implementations.
  ☆46Updated 2 years ago