Red Arrow Documentation

• Contains notes and helper codes on PyTorch Compile, CUDA, and Triton.

Example Code

import torch

def f(x):
    return torch.sin(x)**2 + torch.cos(x)**2