Multi-color Forcing in Graphs. - Chassidy Bozeman; Pamela E. Harris; Neel Jain; Ben Young; Teresa Yu (Graphs Comb., 2020)

Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery. - Yuxin Wen; Neel Jain; John Kirchenbauer; Micah Goldblum; Jonas Geiping; Tom Goldstein (NeurIPS, 2023)

Bring Your Own Data! Self-Supervised Evaluation for Large Language Models. - Neel Jain; Khalid Saifullah; Yuxin Wen; John Kirchenbauer; Manli Shu; Aniruddha Saha; Micah Goldblum; Jonas Geiping; Tom Goldstein (CoRR, 2023)

Baseline Defenses for Adversarial Attacks Against Aligned Language Models. - Neel Jain; Avi Schwarzschild; Yuxin Wen; Gowthami Somepalli; John Kirchenbauer; Ping-yeh Chiang; Micah Goldblum; Aniruddha Saha; Jonas Geiping; Tom Goldstein (CoRR, 2023)

NEFTune: Noisy Embeddings Improve Instruction Finetuning. - Neel Jain; Ping-yeh Chiang; Yuxin Wen; John Kirchenbauer; Hong-Min Chu; Gowthami Somepalli; Brian R. Bartoldson; Bhavya Kailkhura; Avi Schwarzschild; Aniruddha Saha; Micah Goldblum; Jonas Geiping; et al. (ICLR, 2024)

Be like a Goldfish, Don't Memorize! Mitigating Memorization in Generative LLMs. - Abhimanyu Hans; John Kirchenbauer; Yuxin Wen; Neel Jain; Hamid Kazemi; Prajwal Singhania; Siddharth Singh; Gowthami Somepalli; Jonas Geiping; Abhinav Bhatele; Tom Goldstein (NeurIPS, 2024)

Transformers Can Do Arithmetic with the Right Embeddings. - Sean McLeish; Arpit Bansal; Alex Stein; Neel Jain; John Kirchenbauer; Brian R. Bartoldson; Bhavya Kailkhura; Abhinav Bhatele; Jonas Geiping; Avi Schwarzschild; Tom Goldstein (NeurIPS, 2024)

Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers. - Siddharth Singh; Prajwal Singhania; Aditya Ranjan; John Kirchenbauer; Jonas Geiping; Yuxin Wen; Neel Jain; Abhimanyu Hans; Manli Shu; Aditya Tomar; Tom Goldstein; Abhinav Bhatele (SC, 2024)

GenQA: Generating Millions of Instructions from a Handful of Prompts. - Jiuhai Chen; Rifaa Qadri; Yuxin Wen; Neel Jain; John Kirchenbauer; Tianyi Zhou 001; Tom Goldstein (CoRR, 2024)

LiveBench: A Challenging, Contamination-Free LLM Benchmark. - Colin White; Samuel Dooley; Manley Roberts; Arka Pal; Benjamin Feuer; Siddhartha Jain 001; Ravid Shwartz-Ziv; Neel Jain; Khalid Saifullah; Siddartha Naidu; Chinmay Hegde; Yann LeCun; et al. (CoRR, 2024)

Refusal Tokens: A Simple Way to Calibrate Refusals in Large Language Models. - Neel Jain; Aditya Shrivastava; Chen Zhu 001; Daben Liu; Alfy Samuel; Ashwinee Panda; Anoop Kumar; Micah Goldblum; Tom Goldstein (CoRR, 2024)

LiveBench: A Challenging, Contamination-Limited LLM Benchmark. - Colin White; Samuel Dooley; Manley Roberts; Arka Pal; Benjamin Feuer; Siddhartha Jain 001; Ravid Shwartz-Ziv; Neel Jain; Khalid Saifullah; Sreemanti Dey; Shubh-Agrawal; Sandeep Singh Sandha; et al. (ICLR, 2025)

Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach. - Jonas Geiping; Sean McLeish; Neel Jain; John Kirchenbauer; Siddharth Singh; Brian R. Bartoldson; Bhavya Kailkhura; Abhinav Bhatele; Tom Goldstein (CoRR, 2025)

Exploiting Sparsity for Long Context Inference: Million Token Contexts on Commodity GPUs. - Ryan Synk; Monte Hoover; John Kirchenbauer; Neel Jain; Alex Stein; Manli Shu; Josue Melendez Sanchez; Ramani Duraiswami; Tom Goldstein (CoRR, 2025)