triton-autodiff

Experiment of using Tangent to autodiff triton

MIT License

Stars
68
View Code on GitHub View on X
Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

Related Projects
shapeguard

ShapeGuard allows you to very succinctly assert the expected shapes of tensors in a dynamic, eins...

04 Sep 2020 8
pytorch2c

A Python module for compiling PyTorch graphs to C

15 Feb 2017 89
Expression

Pragmatic functional programming for Python inspired by F#

23 Sep 2020 405
server

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

04 Oct 2018 7,680
CoLT5-attention

Implementation of the conditionally routed attention in the CoLT5 architecture, in Pytorch

20 Mar 2023 223
triton-transformer

Implementation of a Transformer, but completely in Triton

08 Sep 2021 243
ml-design-patterns

Software Architecture for ML engineers

14 Jun 2021 382
diffusers-torchao

End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 ...

05 Aug 2024 213
transformer-deploy

Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer ...

31 Oct 2021 1,644
payton

Payton! Kickstart any 3D OpenGL + GTK Ideas in a few seconds!

17 Mar 2019 46
flash-attention

Fast and memory-efficient exact attention

19 May 2022 11,791