Skip to content

r-rishabh-j/r-rishabh-j

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

10 Commits
Β 
Β 

Repository files navigation

Hi there πŸ‘‹

Rishabh Jain

MS CS (Machine Learning) @ Columbia University

My research interests are in robot manipulation, computer vision and efficient LLM training and inference.

πŸ”₯ Featured Work

batched_specdec β€” Batched Speculative Decoding Engine
PyTorch implementation with prompt batching, non-uniform acceptance lengths, and parallel draft verification.

distillSpec β€” On-Policy Distillation for Speculative Decoding
Achieved 5% token acceptance gain on GSM8k via Reverse KL distillation. Includes batched verification, KV-caching, and pruning.
β†’ HuggingFace Models

3DEgoACT β€” Viewpoint-Invariant Robot Manipulation
Fuses PointNet 3D encoding with egocentric vision. 70% zero-shot success on perturbed camera views where baseline ACT fails.
β†’ Dataset


Stack: Python, C++, PyTorch, HuggingFace, CUDA, MuJoCo

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors