# first-kernel

A custom kernel for PyTorch.

## Installation

```bash
pip install drbh/first-kernel
```

## Usage

```python
import torch
from first_kernel import first_kernel

# Create input tensor
x = torch.randn(1024, 1024, device="cuda")

# Run kernel
result = first_kernel(x)
```

## Development

### Building

```bash
nix develop
nix run .#build-and-copy
```

### Testing

```bash
nix develop .#test
pytest tests/
```

### Test as a `kernels` user

```bash
uv run example.py
```

## License

Apache 2.0