Alternatives and similar repositories for amplification:

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

• redwoodresearch / interp
  Redwood Research's transformer interpretability tools
  ☆14Updated 2 years ago
• gehring / fax
  ☆80Updated 3 years ago
• jax-ml / coix
  Inference Combinators in JAX
  ☆47Updated 5 months ago
• joschu / jax-exp
  ☆24Updated 6 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• alexander-turner / attainable-utility-preservation
  ☆12Updated 3 years ago
• plai-group / covid
  Modelling epidemiological dynamics and performing inference in these models
  ☆27Updated 3 years ago
• mtensor / rulesynthesis
  Code for "Learning Compositional Rules via Neural Program Synthesis"
  ☆60Updated 4 years ago
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 2 years ago
• cambridge-mlg / EinsumNetworks
  ☆52Updated last year
• talesa / amci
  Amortized Monte Carlo Integration
  ☆15Updated 4 years ago
• Kiv / fancy_einsum
  Einsum with einops style variable names
  ☆16Updated 10 months ago
• srush / learns-dex
  ☆33Updated 4 years ago
• TomFrederik / unseal
  Mechanistic Interpretability for Transformer Models
  ☆50Updated 2 years ago
• Bonnevie / rebar
  A Python implementation of the gradient REBAR estimator.
  ☆46Updated 6 years ago
• rgrosse / csc2541_examples
  ☆29Updated 4 years ago
• probcomp / LLaMPPL
  A domain-specific probabilistic programming language for modeling and inference with language models
  ☆119Updated last year
• UCLA-StarAI / LearnPSDD
  ☆15Updated 6 years ago
• pyro-ppl / pyro-api
  Generic API for dispatch to Pyro backends.
  ☆16Updated 3 years ago
• aboustati / vargrad
  Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference
  ☆12Updated 4 years ago
• g-benton / hessian-eff-dim
  Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited
  ☆37Updated 2 years ago
• alignedai / HappyFaces
  The Happy Faces Benchmark
  ☆15Updated last year
• harvardnlp / strux
  ☆18Updated 3 years ago
• google-research / autoconj
  Recognizing and exploiting conjugacy without a domain-specific language
  ☆36Updated 5 years ago
• paulfchristiano / alba
  Implementation of https://medium.com/ai-control/alba-an-explicit-proposal-for-aligned-ai-17a55f60bbcf
  ☆27Updated 7 years ago
• arranger1044 / probabilistic-circuits
  A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.
  ☆49Updated last year
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆11Updated 4 years ago
• neale / avoiding-side-effects
  Code for reproducing the results from the paper Avoiding Side Effects in Complex Environments
  ☆12Updated 3 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated this week
• fzaiser / nonparametric-hmc
  Implementation of Nonparametric Hamiltonian Monte Carlo
  ☆12Updated 2 years ago