Shiv Mehta

Optimization, Geometry, Inference, and Interpretability Emergent Complexity

Philosophy of Study

Pre-undergrad work and writing lives here: archive.shivmehta.com

Worth your time:

Recent History

  1. Graviton Research Capital logo

    Graviton Research Capital

    Quantitative Research InternJun 2026Aug 2026

  2. UCSD Halıcıoğlu Data Science Institute logo

    UCSD Halıcıoğlu Data Science Institute

    Undergraduate Researcher2026Present

    Studying feature representations and latent geometry across diffusion timesteps with Dr. Yusu Wang, and steering methods for more controllable, efficient diffusion models.

  3. Astera Holdings logo

    Astera Holdings

    Quantitative Research InternJun 2025Aug 2025

    Built BART/XGBoost ensembles for MLB outcomes and player props with time-split validation and online Bayesian fair-value updates; a Kelly-criterion strategy reached <0.2 Brier.

  4. SEELab, UCSD CSE Department logo

    SEELab, UCSD CSE Department

    Undergraduate Researcher20252026

    Improved numerical reasoning in quantized LLMs via chain-of-thought over sensor time-series, and built a DPO-tuned multi-agent pipeline with a Bayesian-signal grader under Dr Tajana Rosing and Dr Xiaofan Yu.

  5. Multi Commodity Exchange (MCX) logo

    Multi Commodity Exchange (MCX)

    Machine Learning InternJun 2024Aug 2024

    Kernel-PCA factor models on commodity futures (60% dimensionality reduction), served through a PyTorch inference pipeline on a low-latency C++ (Treelite) backend.

  6. bundl.ai logo

    bundl.ai

    Machine Learning InternJun 2023Aug 2023

    Built a RAG pipeline for a Llama model with hybrid vector/graph retrieval (PyTorch, Neo4j) and cut inference time 40% via distributed fine-tuning and TensorRT.

  7. University of California, San Diego logo

    University of California, San Diego

    B.S. in Data Science (Machine Learning & AI)2023Jun 2027

    Coursework spanning linear algebra, vector calculus, probability & statistics, signal processing, spectral methods, stochastic processes, data structures & algorithms, and graph theory.

Highlighted Work

RoPE in the Residual Stream

20262026

RoPE in the Residual Stream preview

Tracked RoPE's implicit positional manifold through the residual stream of Qwen2.5-3B: a multiscale, Fourier/RoPE-like geometry that is approximately linearly transported (rotated, stretched, and occasionally reorganized) across depth.

Partition Geometry

May 2026Present

Partition Geometry preview

A framework comparing trees, ReLU networks, and polyhedra on shared labeled grids via alignments; proved a general profile lower bound that gives exact tree complexities.

Spectra

Jan 2026Present

[]\begin{bmatrix} \color{#a5cfaf}{\blacksquare} & & & & \\ & \color{#a5bccf}{\blacksquare} & & & \\ & & \color{#aea5cf}{\blacktriangle} & & \\ & & & \color{#cfa5b3}{\bullet} & \\ & & & & \color{#c7a5cf}{\cdot} \end{bmatrix}

A command-line tool that profiles tensor artifacts (.npy, .npz) for sparsity, entropy, and spectral structure, then routes each to the best compression strategy: quantization, sparse COO storage, or truncated SVD.

Prediction Market: Market Making

Dec 2025Present

Prediction Market: Market Making preview

Market-making BTC prediction markets: $10k+ volume across 100+ markets. Survival analysis for fills, markout estimation through random forest quantile regression (Treelite for inference), all optimized over a thin plate spline for spread, sizing, and timing.

IMC Prosperity

Apr 2025Apr 2025

IMC Prosperity preview

Top 6% (peak rank 22 globally) in IMC Prosperity 3: ETF arbitrage, delta hedging, and volatility modeling.

cuda-toolkit

Jan 2025Present

A GPU autocorrelation and sum-reduction library with 50 to 100× speedups over CPU.

Jane Street Puzzles

Dec 2024Apr 2025

Jane Street Puzzles preview

Verified solutions to the Dec '24, Feb, Mar, and Apr '25 monthly puzzles.

Pandera

2024Present

Visualized implied-volatility surfaces (Newton-Raphson), order-flow ravines, and bond yield curves.