Skip to content

Instantly share code, notes, and snippets.

View varadhjain's full-sized avatar
🙏

Varadh Jain varadhjain

🙏
View GitHub Profile
@varadhjain
varadhjain / grpo_demo.py
Created February 3, 2025 07:14 — forked from willccbb/grpo_demo.py
GRPO Llama-1B
# train_grpo.py
import re
import torch
from datasets import load_dataset, Dataset
from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import LoraConfig
from trl import GRPOConfig, GRPOTrainer
# Load and prep dataset
Verifying my Blockstack ID is secured with the address 1M7sco3kdZy2qiSLBkRWQg2XKJaMDHHvQe https://explorer.blockstack.org/address/1M7sco3kdZy2qiSLBkRWQg2XKJaMDHHvQe
@varadhjain
varadhjain / keybase.md
Created July 9, 2016 04:19
Key base proof

Keybase proof

I hereby claim:

  • I am varadhjain on github.
  • I am varadh (https://keybase.io/varadh) on keybase.
  • I have a public key whose fingerprint is 66DE 7361 62AA 7A83 D416 2A91 B67A 2005 A61F 9C1F

To claim this, I am signing this object: