Alternatives and similar repositories for jax-chacha-prng

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

• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 4 years ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 3 years ago
• lightonai / reinforcement-learning-opu
  Code to perform Model-Free Episodic Control using Aurora OPUs
  ☆17Updated 5 years ago
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆11Updated 4 years ago
• anirudh9119 / walkback_nips17
  Variational Walkback, NIPS'17
  ☆28Updated 7 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• srush / jax-lda
  ☆30Updated 3 years ago
• microsoft / statopt
  Statistical adaptive stochastic optimization methods
  ☆32Updated 5 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• gebob19 / introduction_to_normalizing_flows
  Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'
  ☆27Updated 4 years ago
• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆25Updated 5 years ago
• KarenUllrich / binary-VAE
  A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.
  ☆22Updated 5 years ago
• kazukiosawa / ngd_in_wide_nn
  simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…
  ☆14Updated 4 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 5 years ago
• tansey / functionalmf
  Functional matrix factorization via Bayesian tensor filtering
  ☆13Updated 2 years ago
• harvardnlp / strux
  ☆18Updated 3 years ago
• gehring / fax
  ☆80Updated 3 years ago
• plai-group / covid
  Modelling epidemiological dynamics and performing inference in these models
  ☆27Updated 3 years ago
• AndPotap / halfpres_gps
  ☆16Updated 2 years ago
• franrruiz / uivi
  Code for Unbiased Implicit Variational Inference (UIVI)
  ☆14Updated 6 years ago
• uber-research / vargp
  Variational Auto-Regressive Gaussian Processes for Continual Learning
  ☆21Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• djsutherland / igms
  Implicit generative models and related stuff based on the MMD, in PyTorch
  ☆16Updated 4 years ago
• edwardjhu / improved_wasserstein
  Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"
  ☆14Updated 5 years ago
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 2 years ago
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 4 years ago
• PrincetonLIPS / randomized_telescopes
  Code for "Efficient optimization of loops and limits with randomized telescoping sums"
  ☆28Updated 6 years ago
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆13Updated 3 years ago
• danieljtait / ladax
  Layered distributions using FLAX/JAX
  ☆10Updated 4 years ago
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 5 years ago