Skip to content

tinygrad 0.6.0
Compare
Choose a tag to compare
@geohot geohot released this 26 May 01:02
· 2666 commits to master since this release
v0.6.0
f4f23dc

2516 lines now. Some day I promise a release will make it smaller.

  • float16 support (needed for LLaMA)
  • Fixed critical bug in training BatchNorm
  • Limited support for multiple GPUs
  • ConvNeXt + several MLPerf models in models/
  • More torch-like methods in tensor.py
  • Big refactor of the codegen into the Linearizer and CStyle
  • Removed CompiledBuffer, use the LazyBuffer ShapeTracker
Assets 2