Also tinygrad does this https://github.com/tinygrad/tinygrad