imbue-ai / carbsLinks
Cost aware hyperparameter tuning algorithm
☆173Updated last year
Alternatives and similar repositories for carbs
Users that are interested in carbs are comparing it to the libraries listed below
Sorting:
- ☆285Updated last year
- seqax = sequence modeling + JAX☆168Updated 3 months ago
- 🧱 Modula software package☆300Updated 2 months ago
- ☆121Updated last week
- Efficient baselines for autocurricula in JAX.☆200Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆103Updated last month
- fast + parallel AlphaZero in JAX☆104Updated 10 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆645Updated 2 months ago
- Minimal but scalable implementation of large language models in JAX☆35Updated 2 months ago
- Minimal yet performant LLM examples in pure JAX☆198Updated last month
- Benchmarking Agentic LLM and VLM Reasoning On Games☆205Updated 2 months ago
- Efficient optimizers☆276Updated 3 weeks ago
- A simple library for scaling up JAX programs☆144Updated last week
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆321Updated this week
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆104Updated 10 months ago
- A set of Python scripts that makes your experience on TPU better☆54Updated last month
- ☆119Updated 5 months ago
- Learn online intrinsic rewards from LLM feedback☆44Updated 10 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆679Updated this week
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆69Updated 10 months ago
- Synchronized Curriculum Learning for RL Agents☆114Updated last week
- A simple, performant and scalable JAX-based world modeling codebase☆100Updated last week
- Accelerated minigrid environments with JAX☆152Updated 3 weeks ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆349Updated 4 months ago
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.☆104Updated last month
- supporting pytorch FSDP for optimizers☆83Updated 11 months ago
- Solve puzzles. Learn CUDA.☆64Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆132Updated 10 months ago
- ☆106Updated 3 weeks ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆258Updated last month