Alternatives and similar repositories for twisted-smc-lm:

Users that are interested in twisted-smc-lm are comparing it to the libraries listed below

• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆71Updated 4 months ago
• shikaiqiu / compute-better-spent
  ☆52Updated 5 months ago
• probcomp / hfppl
  Probabilistic programming with HuggingFace language models
  ☆102Updated this week
• berlino / seq_icl
  ☆51Updated 9 months ago
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆27Updated 9 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 4 months ago
• AndPotap / einsum-search
  ☆30Updated 5 months ago
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆25Updated 10 months ago
• abhishekpanigrahi1996 / transformer_in_transformer
  ☆44Updated last year
• probcomp / LLaMPPL
  A domain-specific probabilistic programming language for modeling and inference with language models
  ☆118Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆33Updated 7 months ago
• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆35Updated last year
• AllanYangZhou / universal_neural_functional
  ☆49Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• HKUNLP / diffusion-vs-ar
  [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"
  ☆38Updated last month
• harmonic-ai / datasets
  Harmonic Datasets
  ☆36Updated 8 months ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆107Updated 2 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• google-deepmind / spectral_ssm
  ☆31Updated 11 months ago
• srush / mamba-primer
  ☆37Updated 11 months ago
• jopetty / word-problem
  Experiments on the impact of depth in transformers and SSMs.
  ☆23Updated 4 months ago
• jysohn1108 / Looped-Transformer
  Official implementation of the transformer (TF) architecture suggested in a paper entitled "Looped Transformers as Programmable Computers…
  ☆24Updated last year
• ExtensityAI / benchmark
  Evaluation of neuro-symbolic engines
  ☆34Updated 7 months ago
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆48Updated this week
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated last year
• adamkarvonen / SAE_BoardGameEval
  ☆21Updated last month
• Leiay / looped_transformer
  ☆24Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated last year