google-deepmind / ccbo
This repo contains the code associated to the paper: "Constrained Causal Bayesian Optimization" by Aglietti Virginia, Alan Malek, Ira Ktena, and Silvia Chiappa. International Conference on Machine Learning. PMLR, 2023.
☆16Updated 10 months ago
Alternatives and similar repositories for ccbo:
Users that are interested in ccbo are comparing it to the libraries listed below
- Minimum Description Length probing for neural network representations☆19Updated 2 months ago
- Code associated to papers on superposition (in ML interpretability)☆28Updated 2 years ago
- 🧮 Algebraic Positional Encodings.☆12Updated 3 months ago
- ☆34Updated last year
- Learning Universal Predictors☆73Updated 8 months ago
- Graph neural networks in JAX.☆67Updated 9 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 10 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- Evaluation of neuro-symbolic engines☆35Updated 8 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 3 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- An annotated implementation of the Hyena Hierarchy paper☆32Updated last year
- ☆114Updated last week
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- The Energy Transformer block, in JAX☆57Updated last year
- DISCS: The code base for the Benchmark for Discrete Sampling☆54Updated 7 months ago
- Experiments on the impact of depth in transformers and SSMs.☆24Updated 5 months ago
- ☆26Updated 2 years ago
- ☆31Updated last year
- Official code for paper: Conservative objective models are a special kind of contrastive divergence-based energy model☆14Updated last year
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆18Updated last month
- Details for the Science4Cast competition☆62Updated 2 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 4 months ago
- Clean RL implementation using MLX☆30Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated last year
- T5Patches is a set of tools for fast and targeted editing of generative language models built with T5X.☆12Updated 10 months ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- Official repository for the paper "Exploring the Promise and Limits of Real-Time Recurrent Learning" (ICLR 2024)☆11Updated last year
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- Code for minimum-entropy coupling.☆31Updated 9 months ago