google-research / kauldron
Modular, scalable codebase to train ML models
☆36Updated this week
Related projects: ⓘ
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆58Updated 2 years ago
- Documentation for dynamic machine learning systems.☆26Updated last week
- Automatically take good care of your preemptible TPUs☆28Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Code for minimum-entropy coupling.☆29Updated 2 months ago
- ☆29Updated 2 years ago
- A logical, reasonably standardized, but flexible project structure for conducting ml research 🍪☆14Updated last month
- ☆17Updated 5 months ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- Amos optimizer with JEstimator lib.☆79Updated 4 months ago
- ☆38Updated 8 months ago
- ☆27Updated this week
- Fast training of unitary deep network layers from low-rank updates☆28Updated last year
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆41Updated 8 months ago
- A framework for standardizing evaluations of large foundation models, beyond single-score reporting and rankings.☆28Updated this week
- Understanding how features learned by neural networks evolve throughout training☆30Updated this week
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆31Updated 3 months ago
- some common Huggingface transformers in maximal update parametrization (µP)☆76Updated 2 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- INTeractive learning via REPresentatIon Discovery☆33Updated 3 months ago
- Minimum Description Length probing for neural network representations☆15Updated 11 months ago
- Official code for paper: Conservative objective models are a special kind of contrastive divergence-based energy model☆14Updated last year
- A simple library for scaling up JAX programs☆116Updated last month
- a curated list of the role of small models in the LLM era☆16Updated this week
- Research on Tabular Foundation Models☆21Updated last month
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆41Updated 3 months ago
- ☆91Updated last month
- Google Research☆45Updated last year
- ☆29Updated 2 weeks ago
- Image augmentation library for Jax☆36Updated 5 months ago