Searcharxiv⌕ Search

arXiv subjects

Adeem Jassani

Publications and source records attributed to Adeem Jassani.

4 recordsLinked to original sources

ROCm InferaSim: Cut LLM Serving Trial Costs and Deploy the Best Configuration

Serving an LLM efficiently depends on the engine, how the model is split across GPUs, precision, scheduler settings, speculative decoding, KV-cache management, and prefill-decode disaggregation. The best configuration depends on the model, hardware, workload, and objective. Operators usually find it by trial and error. Exhaustively measuring the candidates in our study would take about 25,000 GPU-hours. ROCm InferaSim predicts serving performance from a few short measurements. Choices such as batch size, prompt length, and GPU count change the amount of work per GPU without changing the kernels. ROCm InferaSim benchmarks each distinct set of kernels once, on at most one node. We call this measurement an anchor. Analytical laws carry that anchor to other deployments that run the same kernels. For multi-node deployments, ROCm InferaSim reduces the configuration to a single-node equivalent, measures it, and scales the result analytically. A simulator replays multi-turn traffic, and a tuning agent searches configurations using the predictions. On AMD Instinct MI355X GPUs with vLLM, SGLang, and ATOM, 11 anchors took 9.1 GPU-hours, against 204 GPU-hours to measure the 226 fixed-length configurations they cover. On 172 configurations, time per output token has 8.1% median error. On a multi-turn coding workload, throughput is within 8%. For 159 disaggregated deployments, none used as an anchor, throughput is within 5.5%. A five-candidate short list contains a deployment within 12% of the best feasible pairing of model and latency target.

cs.AI↗

Improving Knowledge Distillation for BERT Models: Loss Functions, Mapping Methods, and Weight Tuning

The use of large transformer-based models such as BERT, GPT, and T5 has led to significant advancements in natural language processing. However, these models are computationally expensive, necessitating model compression techniques that reduce their size and complexity while maintaining accuracy. This project investigates and applies knowledge distillation for BERT model compression, specifically focusing on the TinyBERT student model. We explore various techniques to improve knowledge distillation, including experimentation with loss functions, transformer layer mapping methods, and tuning the weights of attention and representation loss and evaluate our proposed techniques on a selection of downstream tasks from the GLUE benchmark. The goal of this work is to improve the efficiency and effectiveness of knowledge distillation, enabling the development of more efficient and accurate models for a range of natural language processing tasks.

cs.CL↗

GROWTH on S190426c II: GROWTH-India Telescope search for an optical counterpart with a custom image reduction and candidate vetting pipeline

S190426c / GW190426_152155 was the first probable neutron star - black hole merger candidate detected by the LIGO-Virgo Collaboration. We undertook a tiled search for optical counterparts of this event using the 0.7m GROWTH-India Telescope. Over a period of two weeks, we obtained multiple observations over a 22.1 deg^2 area, with a 17.5% probability of containing the source location. Initial efforts included obtaining photometry of sources reported by various groups, and a visual search for sources in all galaxies contained in the region. Subsequently, we have developed an image subtraction and candidate vetting pipeline with ~ 94% efficiency for transient detection. Processing the data with this pipeline, we find several transients, but none that are compatible with kilonova models. We present the details of our observations, working of our pipeline, results from the search, and our interpretations of the non-detections that will work as a pathfinder during the O4 run of LVK.

astro-ph.HE↗

India's first robotic eye for time domain astrophysics: the GROWTH-India telescope

We present the design and performance of the GROWTH-India telescope, a 0.7 m robotic telescope dedicated to time-domain astronomy. The telescope is equipped with a 4k back-illuminated camera giving a 0.82-degree field of view and sensitivity of m_g ~20.5 in 5-min exposures. Custom software handles observatory operations: attaining high on-sky observing efficiencies (>~ 80%) and allowing rapid response to targets of opportunity. The data processing pipelines are capable of performing PSF photometry as well as image subtraction for transient searches. We also present an overview of the GROWTH-India telescope's contributions to the studies of Gamma-ray Bursts, the electromagnetic counterparts to gravitational wave sources, supernovae, novae and solar system objects.

astro-ph.IM↗