maximus12793

# Validate Attention
import math

from flax.linen.attention import dot_product_attention
import jax
import jax.numpy as jnp
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F

maximus12793

1. Check Python3 root
>>> import sys
>>> import os
>>> sys.executable
'/usr/local/bin/python3'

OR

$ which python3
/usr/local/bin/python3

maximus12793

Compiler Lab: Generating Assembly Code

Introduction

• Assigned: Oct. 14
• Due: Oct. 30, 11:59 PM

Note: Keshav Kini (krkini@utexas.edu) is the lead TA for this lab assignment.