openai / ai-and-efficiencyLinks
Submissions for AI and Efficiency SOTA's
☆56Updated 5 years ago
Alternatives and similar repositories for ai-and-efficiency
Users that are interested in ai-and-efficiency are comparing it to the libraries listed below
Sorting:
- Learned Hyperparameter Optimizers☆59Updated 4 years ago
- Python Research Framework☆106Updated 2 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 3 years ago
- Train very large language models in Jax.☆206Updated last year
- Code for the paper, "Distribution Augmentation for Generative Modeling", ICML 2020.☆126Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆241Updated 2 years ago
- ☆39Updated 2 years ago
- JAX implementation ViT-VQGAN☆59Updated 3 years ago
- Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)☆146Updated 4 years ago
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 2 years ago
- ☆24Updated 6 years ago
- Code for the paper "Understanding RL Vision"☆48Updated 2 years ago
- Scaling scaling laws with board games.☆53Updated 2 years ago
- Jiminy Cricket Environment (NeurIPS 2021)☆25Updated 3 years ago
- ☆38Updated last year
- A dataset of alignment research and code to reproduce it☆77Updated 2 years ago
- Evaluation suite for large-scale language models.☆127Updated 4 years ago
- 🎢 Creating and sharing simulation environments for embodied and synthetic data research☆191Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- A centralized place for deep thinking code and experiments☆86Updated 2 years ago
- ☆69Updated last year
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 4 years ago
- Python library which enables complex compositions of language models such as scratchpads, chain of thought, tool use, selection-inference…☆208Updated 2 months ago
- Experiment code for the ICLR 2020 paper "RTFM: Generalising to New Environment Dynamics via Reading".☆38Updated 3 years ago
- Adapting the AlphaZero algorithm to remove the need of execution traces to train NPI.☆79Updated last year
- Neural Turing Machines in pytorch☆48Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago