Aditya Jain
Software Engineer
AI • Backend • Systems
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
Radiology Imaging Laboratory (opens in a new tab)
Machine Learning Research
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