amjadmajid / BabyTorch
BabyTorch is a minimalist deep-learning framework with a similar API to PyTorch. This minimalist design encourages learners explore and understand the underlying algorithms and mechanics of deep learning processes. It is design such that when learners are ready to switch to PyTorch they only need to remove the word `baby`.
☆27Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for BabyTorch
- Cost aware hyperparameter tuning algorithm☆122Updated 4 months ago
- fast + parallel AlphaZero in JAX☆84Updated 7 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆84Updated last week
- Accelerated minigrid environments with JAX☆117Updated 3 months ago
- Solve puzzles. Learn CUDA.☆60Updated 10 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆84Updated last month
- ☆147Updated 2 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆87Updated 11 months ago
- ☆122Updated this week
- Scalable neural net training via automatic normalization in the modular norm.☆119Updated 2 months ago
- ☆197Updated 3 months ago
- Minimal but scalable implementation of large language models in JAX☆25Updated last week
- Implementation of Diffusion Transformer (DiT) in JAX☆252Updated 5 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆94Updated 3 months ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆197Updated last month
- ☆64Updated last week
- seqax = sequence modeling + JAX☆132Updated 3 months ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code☆26Updated 2 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆203Updated 3 weeks ago
- MoMo: Momentum Models for Adaptive Learning Rates☆16Updated 5 months ago
- Explainable Reinforcement Learning (XRL) Resources☆33Updated last month
- ☆139Updated 2 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆63Updated 2 months ago
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆80Updated 10 months ago
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆48Updated last year
- Efficient baselines for autocurricula in JAX.☆173Updated 2 months ago
- A puzzle to learn about prompting☆120Updated last year
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆112Updated 6 months ago
- A simple library for scaling up JAX programs☆125Updated last week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆118Updated 3 months ago