Aditya Jain

Software Engineer

AIBackendSystems

Journey, 2023 to now

Research · Hackathons · Early internships

Swif AI (opens in a new tab)

Identity Platform · AI agents

Swif AI (opens in a new tab)

Security Platform · MDM

Thales Avionics (opens in a new tab)

Software Engineer Intern

Apple MLX (opens in a new tab)

Open source · 10+ merged PRs

Amazon Web Services (opens in a new tab)

Open source · 20+ merged PRs

LanceDB (opens in a new tab)

Open source · 3+ merged PRs

Education

University of California, Irvine, B.S. Computer Science, Intelligent Systems, Dean’s List, Expected June 2027. Previously Irvine Valley College.

Experience

Thales Avionics: Software Engineer Intern, June 2026 – September 2026

  • Developed a Python ILP algorithm to identify reusable In-Flight Entertainment parts, reducing manual audit time by ~40%
  • Processed 10TB of aircraft BITE data with Databricks and PySpark to support fleet-wide predictive maintenance at scale
  • Designed PySpark feature pipelines with Delta Lake and Databricks Feature Store to prepare aircraft data for ML workloads
  • Built 4 interactive Databricks dashboards used by 15+ airlines to surface maintenance insights and repair recommendations

Radiology Imaging Laboratory: Machine Learning Researcher, November 2025 – June 2026

  • Implemented iterative CT reconstruction algorithms to reduce Gaussian noise, improving overall image quality by ~20%
  • Applied CNN models using PyTorch and OpenCV to CT scans to detect coronary artery disease and identify calcification
  • Accelerated CT reconstruction with GPU computing using an in-house CT simulator for faster medical image processing

Swif AI, Security Platform: Returning Software Engineer Intern, September 2025 – December 2025

  • Engineered Go services with MDM and REST APIs, cutting security policy deployment time by ~40% across company devices
  • Built Redis-backed security dashboards with GraphQL APIs tracking compliance and endpoint risk for 1,000+ customers
  • Implemented AI-assisted MDM workflows in Go for 15+ customers, automating security policy generation and deployment

Swif AI, Identity Platform: Software Engineer Intern, June 2025 – September 2025

  • Engineered Python AI agents with Playwright for automated app provisioning, increasing employee onboarding efficiency by 2×
  • Developed Vue.js and GraphQL workflows for software installation and removal across employee onboarding and offboarding
  • Designed self-correcting AI agents with fallback recovery, reducing manual intervention on failed provisioning by ~35%

Open source

Apple MLX: 10+ merged PRs

  • Fixed second-order gradients for 5 MLX ops, enabling higher-order autodiff across a framework with 28K+ stars on GitHub
  • Resolved C++ segfaults and einsum/vmap correctness bugs across 10+ merged PRs, with all fixes validated against NumPy

Amazon Web Services: 20+ merged PRs

  • Closed cross-account safety-check bypass in AWS CDK asset publishing, securing a toolchain with 5M+ weekly npm downloads
  • Reduced AWS CDK refactor runtime by 38% and fixed security group diff blind spots across 20+ merged CDK contributions

LanceDB: 3+ merged PRs

  • Fixed hybrid search ignoring pagination and query parameters in LanceDB’s Python SDK, supporting 7M+ monthly downloads
  • Fixed stale merge-insert delete conditions in LanceDB, preventing accidental row deletions during insert/update operations

Projects

Graphify: Python, Tree-sitter, NetworkX, MCP

  • Built a static analysis tool with Tree-sitter to map large codebases into searchable dependency graphs for AI assistants
  • Reduced code sent to AI assistants by ~70% using NetworkX graph retrieval to find most relevant files and dependencies
  • Designed an MCP server enabling AI assistants to retrieve relevant code, dependencies, and context directly from the graph

CacheFlow: Rust, CUDA C++, gRPC, Tokio, Protobuf

  • Built a GPU KV-cache manager in Rust for LLM inference with paged allocation and prompt-prefix caching across requests
  • Implemented LRU eviction and host-memory paging across 8 CUDA streams for efficient cache reuse under heavy concurrency
  • Connected Rust to CUDA C++ kernels through FFI, reaching 11K block ops/sec at 91µs average latency under cache eviction

Let's build something.