Rare find

nGPT. A PyTorch implementation of Nvidia's normalized GPT that trains AI models using a different mathematical approach.

github.com/lucidrains/nGPT-pytorch

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

June 2025
  • Release —0.2.16
  • 0.2.16
  • address https://github.com/lucidrains/nGPT-pytorch/issues/12
March 2025
  • Release —0.2.15
  • complete ability to breed two normalized transformers, add another fu…
  • Release —0.2.14
  • refactor evo functions to ready for full transformer
  • Release —0.2.12
  • prepare to crossover entire transformer
  • Release —0.2.11
  • evolution will be part of the final equation.
  • update
  • Release —0.2.10
  • cross over the feedforward scales as well
  • Release —0.2.9
  • prepare for breeding the nGPT feedforwards
November 2024
  • Release —0.2.7
  • add enable_cudnn
  • Release —0.2.6
  • fix returning zeros when row is all masked out for non-causal, and re…
  • Release —0.2.5
  • protect against row being all masked out in sdpa
  • oops, got it backwards
  • address https://github.com/lucidrains/nGPT-pytorch/issues/11 with a h…
  • 0.2.2
  • Revert "address https://github.com/lucidrains/nGPT-pytorch/issues/9"
  • address https://github.com/lucidrains/nGPT-pytorch/issues/9
October 2024
  • line things up a bit more accurately for follow up research
  • prep for kv cache
  • fix value residual