Machine Learning Researcher · Patent Technology Specialist at Proskauer Rose
Intrigued by the opportunity to apply his technical skills to the legal domain, Ahmed Taha transitioned from serving as a USPTO Patent Examiner to joining Proskauer Rose as a Technology Specialist. He is currently pursuing an M.S. in Computer Science at Columbia University, where his research focuses on agentic AI safety/alignment and computer vision for medical imaging.
COLM 2026 Workshop on Efficient Reasoning · To appear · Cited by 0
TL;DREvaluates frozen anti-sycophancy system prompts under both valid corrections and misleading user pressure on SycoBench-600 across Phi-3-mini, Llama-3.1-8B, and Mistral-7B — showing that lower reversal rates can mask blanket answer inertia and that no prompt improves correction selectivity on all three models.
Preprint, 2026 · Under review · Cited by 0
TL;DRPaired counterfactual benchmark that audits whether nine frozen vision-language models change their spinal-radiology reports when apparent age and sex are edited while target pathology is preserved — measurable recommendation drift in all nine, with management recommendations less stable than diagnostic-label overlap under the same demographic edit.
Preprint, 2026 · Under review · Cited by 0
TL;DRBenchmark and simulated EHR environment of 840 clinical scenarios in 420 matched pairs that tests whether medical LLM agents preserve documentation integrity when institutional context rewards shortcuts — across seven agents, task completion and documentation integrity diverge, and a pre-submission compliance check reduces discrepancies at a measurable cost to task completion (four-reviewer validation, Fleiss' κ = 0.905).
Browse the full publications page, with abstracts and citations →
Counterfactual benchmark auditing demographic sensitivity in spinal radiology vision-language model reports.
Benchmark and simulated EHR environment auditing documentation integrity in medical LLM agents under institutional pressure.
Multi-model LLM router with query complexity classification, dissent detection, and semantic caching.
PyTorch implementation of Mixture-of-Recursions: recursive transformers with dynamic per-token routing (NeurIPS 2025).
JAX/Flax implementation of Relaxed Recursive Transformers with layer-wise LoRA (ICLR 2025).
Standardized CUDA kernel annotations (#6532) by applying the existing CUML_KERNEL macro to the 16 remaining C++ kernels; six of them gain internal linkage, so libcuml.so no longer exports externally linked kernel symbols, as confirmed by RAPIDS detect-weak-linking.
Added GDB and LLDB pretty-printers for cuda::hierarchy in libcu++ (#10089) that report each level's bottom unit and dimensions, reading static extents from type information and dynamic extents from stored values without calling into the debugged program; tests cover partial hierarchies, derived descriptors, and live updates.
Made CUB's DeviceReduce::ReduceByKey run-to-run deterministic (#9995): the public dispatch now opts into the stable cross-tile ordering already supported by TilePrefixCallbackOp, so FP64 segments spanning multiple tiles combine their partials in a fixed order and identical calls return bit-identical aggregates, while RLE and the deprecated dispatcher keep their previous behavior; a regression test checks the reported reproducer bit-for-bit across repeated calls.
Made --stopping-criterion reject conflicting options (#441) that would silently discard parameters or overwrite a criterion set earlier in the same scope, while still letting benchmark-local options override inherited global settings.
Reworked per-run MSA output handling (#31) in OpenFold-3 so each run gets a unique temporary workspace: raw ColabFold records are saved as soon as the server response validates and OpenFold-formatted alignments once processing finishes, with new save_openfold_outputs, save_colabfold_outputs, and colabfold_output_dir settings, plus cleanup that can no longer delete explicit output directories.
Moved OpenFold-3 inference intermediates (#342) out of /tmp and under <output_dir>/openfold3_intermediates, so temporary MSA and template files left by a crashed cluster job stay next to that job's outputs; directories created by a run are removed after normal completion or a handled error.
Annotated OpenFold-3 inference configurations (#342), split out of #363 at a maintainer's request: typed InferenceDataModule.inference_config as InferenceJobConfig, narrowed InferenceExperimentRunner to InferenceExperimentConfig, and removed the 17 matching mypy-baseline.txt entries without changing runtime behavior.
Added an optional residue_name for SMILES ligands (#323) in OpenFold-3 without changing the LIG# defaults: explicit names are trimmed, uppercased, and validated as ASCII alphanumeric, rejected on collisions with CCD, non-canonical, or standard polymer residue names, and carried into the SMILES component mapping, with full names written to mmCIF and three-character truncation documented for legacy PDB output.
Fixed assigned C=C planarity on OpenFold-3's default inference path (#136): RDKit-assigned E/Z, non-aromatic sp2 C=C bonds are detected and their heavy-atom dihedrals pulled to the assigned cis/trans plane by a flat-bottom correction late in diffusion, with one more projection after rollout so the sampler cannot undo it, while queries without matching bonds are left unchanged; the worst targeted dihedral deviation fell from 101.13° to 15.01°.
Stopped compiled Domino cross entropy from materializing full-vocabulary logits (#190718): the AOTAutograd partitioner now recomputes the cheap cat/log-softmax path in backward while keeping small row statistics and expensive projections saved; on the reported BF16 shape, compiled time drops from 5.25 ms to 3.69 ms and temporary memory from 1,940 MiB to 0.1 MiB.
Let the Inductor scheduler fuse compatible sibling CUDA reductions (#190718) when shared direct reads cover at least half of the larger reduction's input traffic, with indirect reads and disjoint slices excluded; split out of #193121 at a maintainer's request, and on the reported BF16 shape it cuts the vocabulary kernels from three to two, taking compiled time from 3.69 ms to 2.64 ms.
Added complex-input support to jax.scipy.special.gamma using the Lanczos approximation and reflection formula in Google JAX.
Made image resizing a JAX primitive (#8452): resize operations now lower through a dedicated image_resize primitive that keeps identity resizes as no-ops and uses the existing implementation as its default lowering, while preserving batching, autodiff, explicit sharding, Checkify, Jet, key-reuse, and BCOO sparsify behavior.
August 2026 - Present
Supporting patent prosecution and litigation.
August 2025 - May 2027
(Thesis track)
August 2025 - Present
scholar.google.com/citations?user=FQ1XbKcAAAAJ
— Currently developing an RL policy to enhance AI-assisted tumor segmentation for radiation imaging under Dr. Yading Yuan
— Led the development of SpineFairBench and MedInsider; under review as first author
January 2025 - August 2026
GS-11 Patent Examiner · January 2026 - August 2026
— Evaluated 50+ machine learning and imaging patent applications, determining novelty and patentability through comprehensive prior art searches
— AU: 2613
GS-9 Patent Examiner · January 2025 - January 2026
— Awarded commendation letter from OPQA Director for office action quality
August 2024 - August 2026
June 2020 - January 2025
Led full-stack development projects generating over seven figures in revenue. Built patient-management systems, auto-dealer platforms, and restaurant applications. Registered trademark for company slogan.
2018 - 2024
— Senior Capstone: Adapted Strength - Contributions Github
— D2 Wrestling Sports Club
August 2017 - February 2021
Sales Manager · Full-time · August 2018 - February 2021
Sales Specialist · Part-time · August 2017 - August 2018