amjadmajid / BabyTorchLinks
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`.
☆26Updated 3 weeks ago
Alternatives and similar repositories for BabyTorch
Users that are interested in BabyTorch are comparing it to the libraries listed below
Sorting:
- Implementation of Diffusion Transformer (DiT) in JAX☆278Updated last year
- ☆126Updated last month
- ☆270Updated 11 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆87Updated 3 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆95Updated last week
- ☆104Updated 2 weeks ago
- Efficient optimizers☆220Updated last week
- ☆80Updated 7 months ago
- ☆65Updated last year
- Solve puzzles. Learn CUDA.☆64Updated last year
- Accelerated minigrid environments with JAX☆139Updated 2 weeks ago
- ☆150Updated 10 months ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆126Updated last month
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆20Updated 4 months ago
- Minimal but scalable implementation of large language models in JAX☆35Updated 7 months ago
- Benchmarking Agentic LLM and VLM Reasoning On Games☆156Updated last month
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆92Updated 5 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆134Updated last week
- PyTorch Single Controller☆231Updated this week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆122Updated 11 months ago
- A simple library for scaling up JAX programs☆139Updated 7 months ago
- Cost aware hyperparameter tuning algorithm☆158Updated 11 months ago
- Implementation of Soft Actor Critic and some of its improvements in Pytorch☆60Updated 4 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆102Updated 9 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆59Updated 8 months ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Updated 8 months ago
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆118Updated last year
- A functional training loops library for JAX☆88Updated last year
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆103Updated 2 months ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆57Updated last year