Downloads · 30 days
0
cahlen/zaremba-cayley-cuda
zaremba-cayley-cuda is a machine learning model from cahlen. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for kernels. The card lists the license as mit.
Computes Cayley graph diameters for the Zaremba generators {1,...,5} acting on SL(2, Z/pZ) via GPU-parallel BFS.
Downloads · 30 days
0
Access
Public
Updated Apr 14, 2026
Repo size
625 KB
Likes
0
Public
Click a slice to open those files.
.so342 KB · 95%
From the Hugging Face model README
Computes Cayley graph diameters for the Zaremba generators {1,...,5} acting on SL(2, Z/pZ) via GPU-parallel BFS.
import torch
from kernels import get_kernel
kernel = get_kernel("cahlen/zaremba-cayley-cuda")
result = cayley.diameter(primes)
nvcc -O3 -arch=sm_90 -o zaremba_cayley cayley_diameter/cayley_gpu.cu -lm
All computation results are open:
@misc{humphreys2026bigcompute,
author = {Humphreys, Cahlen},
title = {bigcompute.science: GPU-Accelerated Computational Mathematics},
year = {2026},
url = {https://bigcompute.science}
}
Human-AI collaborative. Not peer-reviewed. All code and data open.