Pinned Loading

1. gpu-mem-calculator gpu-mem-calculator Public

   GPU Memory Calculator for LLM Training - Calculate GPU memory requirements for training Large Language Models with support for multiple training engines including PyTorch DDP, DeepSpeed ZeRO, Megat…

   Python

2. build_nanogpt build_nanogpt Public

   Reproduction of GPT2 following GPT2 paper and GPT3 paper as well as tutorial from Anjrey Karpathy.

   Jupyter Notebook

3. SOM-in-JAX SOM-in-JAX Public

   Self-Organizing Maps implemented in JAX

   Python

4. vlm vlm Public

   PyTorch implementation of PaliGemma Vision Language Model

   Python

5. neural_active_inference neural_active_inference Public

   Official implementation of paper "A neural active inference model of perceptual-motor learning" published on Computational Neuroscience in 2023.

   Python 7

6. active_inference active_inference Public

   Playground for active inference in Python

   Python 2