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 8 months ago
Alternatives and similar repositories for ccbo:
Users that are interested in ccbo are comparing it to the libraries listed below
- Experiments on GPT-3's ability to fit numerical models in-context.☆14Updated 2 years ago
- Evaluation of neuro-symbolic engines☆34Updated 6 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 8 months ago
- ☆38Updated last week
- JAX/Flax implementation of the Hyena Hierarchy☆33Updated last year
- ☆31Updated 2 years ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆18Updated last month
- ☆34Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Graph neural networks in JAX.☆67Updated 8 months ago
- Neural Networks for JAX☆83Updated 4 months ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- ☆34Updated 2 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆40Updated 8 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆87Updated 8 months ago
- Official implementation of the paper "Interventions, Where and How? Experimental Design for Causal Models at Scale", NeurIPS 2022.☆19Updated 2 years ago
- Efficient Computation of d-Dimensional Earth Mover's Distance☆9Updated 10 months ago
- Minimum Description Length probing for neural network representations☆18Updated 3 weeks ago
- A domain-specific probabilistic programming language for modeling and inference with language models☆114Updated last year
- An annotated implementation of the Hyena Hierarchy paper☆32Updated last year
- T5Patches is a set of tools for fast and targeted editing of generative language models built with T5X.☆12Updated 8 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated last year
- Unofficial implementation of Conformal Language Modeling by Quach et al☆29Updated last year
- Modular, scalable library to train ML models☆60Updated this week
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 months ago
- some common Huggingface transformers in maximal update parametrization (µP)☆78Updated 2 years ago
- ☆111Updated 2 weeks ago