Work

Achievements & Honors

Google DeepMind APAC Research Symposium 2026

Selected as a student research delegate.

2026
EY Techathon 6.0

Secured a Top-3 finish among 180,000+ participants nationwide.

2026
Amazon ML Challenge 2025

Ranked 66th out of 82,789 participants in a national ML competition.

2025
STC-UP Hackathon

Ranked Top 11 across India; only undergraduate team competing against industry startups.

2025

Research & Professional Experience

Research Intern

Causal Evidence & Decisions Studio, Prof. Harsh Parikh – Yale University
May 2026 – Jul 2026
New Haven, USA
  • Formulating a theoretical framework to construct Confidence Sets of Representations.
  • Implemented and evaluated multiple causal representation learning algorithms, applying structural relaxations to bound the unidentifiable representation space.
  • Overcoming the limitations of purely predictive models by recovering true causal mechanisms, enabling robust counterfactual reasoning and scientific discovery.

Undergraduate Research Assistant

P-Square Lab, Prof. Parikshit Pareek – IIT Roorkee
Sep 2025 – Present
Roorkee, India
  • Demonstrated orthogonal task structures induce disentanglement in MLPs, enabling surgical unlearning. [Read Paper]
  • Designed amortized inference for zero-shot Gaussian Process kernel discovery from limited data using PFNs.
  • Engineered a sparsity-aware latent diffusion model for ultra-sparse jet image synthesis in high-energy physics (CERN). [Read Paper]

Research Engineer

Elimentary (YC F26)
Nov 2025 – Jan 2026
Bangalore, India
  • Built secure, quantized local LLM APIs, achieving granular layer-wise observability.
  • Optimized GPU throughput by implementing sequence scheduling for concurrent LLM inference requests.
  • Designed a no-code LLM evaluation platform integrated with Langfuse and Grafana for comprehensive monitoring.

Publications

  1. bilinear_mlps.png
    Structural Disentanglement in Bilinear MLPs via Architectural Inductive Bias
    Ojasva Nema, Kaustubh Sharma, Aditya Chauhan, and Parikshit Pareek
    ICLR Under Review Under review at ICLR 2027 2026
    Architectural inductive bias supports structural disentanglement which, in turn, helps unlearning and generalization.
  2. pfns_spectral.png
    Mechanistic Evidence for Spectral Structures in Prior-Data Fitted Networks
    Ojasva Nema, Kaustubh Sharma, Srijan Tiwari, and Parikshit Pareek
    ICML Workshop ICML Mech. Interp. workshop 2026
    An interpretability-driven framework for amortized spectral discovery from pre-trained PFNs with decoupled attention.
  3. mlp_vs_attn.png
    Localised Routing, Distributed Execution: How LLMs Process Semantic Domains
    Ojasva Nema, Kaustubh Sharma, Abhiraj Bharangar, Manjot Singh, and Srijan Tiwari
    ICLR Under Review Under review at ICLR 2027 2026
    Investigating how LLMs internally represent and process diverse domain knowledge. Found that attention routes domain identity while MLPs store domain-specific logic.
  4. Image-alchemy.png
    Image-Alchemy: Personalized Text-to-Image Generation
    Ojasva Nema, Kaustubh Sharma, Cherish Puniani, and Amritanshu Tiwari
    ICLR Workshop ICLR DeLTa Workshop 2025
    A personalization method for diffusion models advancing subject fidelity in text-to-image generation.
  5. sparsity_aware.png
    Sparsity-Aware Representation Learning for Jet Image Generation via Guided Latent Diffusion
    Ojasva Nema, Kaustubh Sharma, and Parikshit Pareek
    Preprint Preprint 2025
    Modeled a physics-aware diffusion model for jet image synthesis at CERN’s Large Hadron Collider.