Skip to content
View r-rishabh-j's full-sized avatar
πŸ”
Learning
πŸ”
Learning

Highlights

  • Pro

Organizations

@ITBench-NFR

Block or report r-rishabh-j

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Maximum 250 characters. Please don't include any personal information such as legal names or email addresses. Markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
r-rishabh-j/README.md

Hi there πŸ‘‹

Rishabh Jain

MS CS (Machine Learning) @ Columbia University

I build systems that make ML inference faster. My research interests are in efficient LLMs training and inferece, computer vision and robotics.

πŸ”₯ 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

Pinned Loading

  1. batched_specdec batched_specdec Public

    Batched Implementation of Speculative Decoding with a Draft Model (Leviathan et al. 2023).

    Python

  2. distillSpec distillSpec Public

    Using On-Policy Knowledge Distillation for Accelerating Speculative Decoding

    Python

  3. 3DEgoACT 3DEgoACT Public

    Combining 3D point clouds with egocentric features for Robot Manipulating

    Python 1

  4. ITBench-NFR/itbench-nfr ITBench-NFR/itbench-nfr Public

    ITBench NFR scripts

    Python

  5. MAGIC-TBR MAGIC-TBR Public

    Forked from surbhimadan92/MAGIC-TBR

    This is the official Repository containing the implementaton of the paper titled: "MAGIC-TBR: Multiview Attention Fusion for Transformer-based Bodily Behavior Recognition in Group Settings"

    Jupyter Notebook