Anselm Levskaya
Staff Research Engineer at Google
San Francisco, California, United States
Summary
🤩
Rockstar🎓
Top SchoolAnselm Levskaya is a Staff Research Engineer in San Francisco with 15 years of experience building and scaling numerical programming and ML infrastructure at Google, focused on JAX and Gemini. He combines deep numerical expertise—fixing nondeterminism in einsum and implementing JVP rules for eigh in JAX—with practical systems work across distributed optimizers, Flax, and T5X for large-scale language model training. His background spans from founding-scientist roles in synthetic biology and genomics to hands-on AI research, giving him a rare mix of wet-lab experimental rigour and production ML engineering. He’s an active open-source contributor to prominent projects (JAX, TensorFlow Probability, Flax, Certbot) and has driven infrastructure and testing automation for reproducible deployments. Colleagues rely on him to untangle subtle numerical and type-stability bugs that surface only at scale. He often bridges research and production, turning algorithmic innovations into robust, tested code used across Google's ML stack.
15 years of coding experience
9 years of employment as a software developer
Bachelor of Science (B.S.) Physics, Bachelor of Science (B.S.) Physics at Cornell UniversityDoctor of Philosophy (PhD) Biophysics, Doctor of Philosophy (PhD) Biophysics at University of California, San Francisco
Github Skills (44)
transformers10
javascript10
assembly10
distributed-training10
python10
optim10
testing10
cpu10
memory-management10
machine-learning10
bash10
cicd10
flax10
numpy10
x8610
Programming languages (11)
Github contributions (5)
levskaya/jslinux-deobfuscated
Dec 2011 - Jan 2019
An old version of Mr. Bellard's JSLinux rewritten to be human readable, hand deobfuscated and annotated.
Role in this project:
Back-end Developer
Back-end DeveloperContributions:68 commits, 2 PRs, 4 pushes in 7 years 2 months
Contributions summary:Anselm appears to be refactoring and deobfuscating code within the JSLinux emulator, focusing on the core CPU emulation routines. They have rewritten parts of the PCEmulator and CPU_X86 code, including the addition of comments and the renaming of symbols. The user's work involves significant changes to fundamental components related to memory access, interrupt handling, and the overall execution flow of the emulator. Additionally, they added the original CPU code for reference.
rewrittenreadablehuman-readabledeobfuscatedannotated
google/flax
Feb 2020 - Jan 2023
Flax is a neural network library for JAX that is designed for flexibility.
Role in this project:Back-end Developer & DevOps Engineer
Back-end Developer & DevOps EngineerContributions:2 releases, 368 reviews, 213 commits in 2 years 11 months
Contributions summary:Anselm contributed to the improvement of the input pipeline for the lm1b dataset, specifically focusing on adding dynamic batching capabilities to enhance the efficiency of the language model training. The user also made several layout and fix improvements across various example files and included general bug fixes and integration of checkpointing functionality. Additional work included merging branch changes from the prerelease branch and incorporating updates.
deep-learningneural-networksneural-networkflaxjax
Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.