Triton Transformer. A transformer neural network written in a faster, lower-level programming language.

github.com/lucidrains/triton-transformer

Vaya's read on this project

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

Updates

April 2022
  • Release —0.1.1
  • extraneous layernorm
  • make sure example works
  • Release —0.1.0
  • remove all biases, as 540 billion parameter PaLM model shows it is no…
March 2022
  • Release —0.0.28
  • pin to last working version of triton
January 2022
  • todo
November 2021
  • Release —0.0.27
  • add stable layernorm, from cogview paper
  • Release —0.0.26
  • fix bug with triton layernorm
October 2021
  • cleanup
  • DRY a little
  • Release —0.0.25
  • directly calculate the causal mask within the triton softmax kernel
  • explain loss calculation
  • readme
  • todo check
  • Release —0.0.24
  • make sure dropout backwards is correct
  • Release —0.0.23
  • follow Phil Ts advice
  • Release —0.0.22
  • Release —0.0.21
  • add memory efficient dropout from triton lang tutorials
September 2021
  • move images to folder
  • break components into individual files
  • Update README.md
  • softmax forwards and backwards results