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 10 months ago
Alternatives and similar repositories for BabyTorch:
Users that are interested in BabyTorch are comparing it to the libraries listed below
- ☆87Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆169Updated last week
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆81Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆76Updated 3 weeks ago
- Implementation of Diffusion Transformer (DiT) in JAX☆270Updated 9 months ago
- 🧱 Modula software package☆187Updated this week
- ☆150Updated 7 months ago
- supporting pytorch FSDP for optimizers☆80Updated 3 months ago
- ☆87Updated 2 weeks ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆103Updated 4 months ago
- ☆27Updated 8 months ago
- ☆152Updated last year
- Cost aware hyperparameter tuning algorithm☆148Updated 9 months ago
- Solve puzzles. Learn CUDA.☆63Updated last year
- Custom triton kernels for training Karpathy's nanoGPT.☆18Updated 5 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 10 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆108Updated 3 months ago
- ☆91Updated 2 months ago
- Accelerated minigrid environments with JAX☆132Updated 8 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 8 months ago
- A simple library for scaling up JAX programs☆134Updated 5 months ago
- seqax = sequence modeling + JAX☆151Updated 2 weeks ago
- Accelerated First Order Parallel Associative Scan☆180Updated 7 months ago
- ☆215Updated 8 months ago
- fast + parallel AlphaZero in JAX☆94Updated 3 months ago
- Codebase to fully reproduce the results of "No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPO" (M…☆25Updated 4 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Efficient optimizers☆185Updated this week
- A package for defining deep learning models using categorical algebraic expressions.☆60Updated 8 months ago
- ☆76Updated 8 months ago