Jackson Kunde
Hi, I am a senior at the University of Wisconsin, Madison, studying mathematics and computer science.
Current Work
University of Wisconsin
September 2023 - PresentI am researching large language models in Professor Kangwook Lee’s lab.
AI & Machine Learning Intern at Ohalo
June 2024 - PresentI also work on Reinforcement Learning for Robotics at Ohalo.
Wisconsin AI Safety Initiative
September 2023 - PresentI serve as the Deputy Director of the Wisconsin AI Safety Initiative.
For all of my work experience, see my resume.
Project Highlights
Testing Generalization Bounds: How Good Is Theory in Practice?
February 2024Here, I test how the mathematical theory of generalization bounds works in practice.
Language Transformer Implementation
November 2024Here, I implement a decoder-only language model using Pytorch, and I build a trainer to train the model on a shakespeare dataset.
Proximal Policy Optimization Implementation
February 2024Here, I implement PPO and show it working for breakout and cartpole.
Reinforcement Learning from Human Feedback Implementation
November 2024Here, I implement RLHF using my own transformer implementation and a simple policy gradient algorithm.
For details on my projects, see the projects page.