Productive, portable, and performant GPU programming in Python.
Repository Index
47,133 repositories indexed
1–4 of 4 · Most stars
Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...
DiffSharp: Differentiable Functional Programming
Machine learning compiler in Go. A from-scratch tinygrad port: graph-rewrite IR, autodiff as a compiler pass, zero-CGO WebGPU backend.