imbue-ai / carbsLinks
Cost aware hyperparameter tuning algorithm
☆171Updated last year
Alternatives and similar repositories for carbs
Users that are interested in carbs are comparing it to the libraries listed below
Sorting:
- ☆283Updated last year
- seqax = sequence modeling + JAX☆168Updated 3 months ago
- 🧱 Modula software package☆291Updated 2 months ago
- Minimal yet performant LLM examples in pure JAX☆186Updated last month
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆99Updated 3 weeks ago
- ☆116Updated 3 weeks ago
- Efficient baselines for autocurricula in JAX.☆196Updated last year
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆317Updated this week
- A simple library for scaling up JAX programs☆144Updated 11 months ago
- Efficient optimizers☆274Updated last week
- Minimal but scalable implementation of large language models in JAX☆35Updated last month
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆166Updated 3 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆671Updated this week
- Learn online intrinsic rewards from LLM feedback☆43Updated 10 months ago
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Updated last year
- supporting pytorch FSDP for optimizers☆83Updated 10 months ago
- Accelerated minigrid environments with JAX☆151Updated this week
- fast + parallel AlphaZero in JAX☆103Updated 10 months ago
- A set of Python scripts that makes your experience on TPU better☆54Updated last month
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆639Updated 2 months ago
- Synchronized Curriculum Learning for RL Agents☆114Updated last month
- Benchmarking Agentic LLM and VLM Reasoning On Games☆202Updated 2 months ago
- ☆120Updated 4 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆344Updated 3 months ago
- PyTorch-native post-training at scale☆83Updated this week
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.☆103Updated 3 weeks ago
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆104Updated 9 months ago
- Solve puzzles. Learn CUDA.☆64Updated last year
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆255Updated last month
- Jax/Flax rewrite of Karpathy's nanoGPT☆62Updated 2 years ago