google-deepmind / alphadev
☆702Updated last year
Alternatives and similar repositories for alphadev:
Users that are interested in alphadev are comparing it to the libraries listed below
- Monte Carlo tree search in JAX☆2,427Updated 2 months ago
- ☆2,720Updated 9 months ago
- Reinforcement learning environments for compiler and program optimization tasks☆932Updated 4 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆883Updated last year
- ♟️ Vectorized RL game environments in JAX☆440Updated this week
- Convolutions for Sequence Modeling☆876Updated 8 months ago
- ☆521Updated last year
- Evolution Through Large Models☆711Updated last year
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,251Updated 2 months ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆827Updated 4 months ago
- AlphaZero in JAX☆74Updated 10 months ago
- Really Fast End-to-End Jax RL Implementations☆810Updated 5 months ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,966Updated this week
- Pytorch Implementation of MuZero☆348Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,733Updated 2 months ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆478Updated 2 weeks ago
- Automatic gradient descent☆207Updated last year
- Open weights language model from Google DeepMind, based on Griffin.☆620Updated 7 months ago
- ☆454Updated 2 years ago
- ☆873Updated 7 months ago
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…☆156Updated 3 years ago
- Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.☆657Updated this week
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆164Updated last year
- ☆348Updated this week
- Language Modeling with the H3 State Space Model☆516Updated last year
- A structured implementation of MuZero☆207Updated 2 years ago
- Public repo for the NeurIPS 2023 paper "Unlimiformer: Long-Range Transformers with Unlimited Length Input"☆1,059Updated 11 months ago
- ☆864Updated last year
- Neural Networks and the Chomsky Hierarchy☆201Updated 10 months ago
- RL Environments in JAX 🌍☆703Updated 7 months ago