Jonas Köhler - Member Of Technical Staff at CuspAI

Jonas Köhler

Member Of Technical Staff at CuspAI

Berlin, Germany

Join Prog.AI to see contacts

Summary

🤩

Rockstar

🎓

Top School

Jonas Köhler is a machine learning researcher-engineer with 11 years of experience, currently a Member of Technical Staff at CuspAI working on ML for accelerated material discovery. He has held research roles at Microsoft on deep-learning augmented ab-initio electronic structure and at Freie Universität Berlin on normalizing flows and diffusion models for molecular sampling, plus an internship on AlphaFold at DeepMind. Jonas combines strong theoretical grounding (MSc AI cum laude, extended PhD work in ML/applied math and a UCLA IPAM fellowship) with practical open-source contributions—e.g., improving the S2CNN spherical CNN examples used in equivariant learning. Based in Berlin, he blends academic rigor with hands-on engineering across physics-aware ML, symmetry-preserving architectures, and production-oriented research. An understated strength is his track record of refactoring and simplifying research codebases to make advanced models more reproducible and usable.

11 years of coding experience

10 years of employment as a software developer

Doktor (Ph.D.) - unfinished, Machine Learning / Applied Mathematics, Doktor (Ph.D.) - unfinished, Machine Learning / Applied Mathematics at Freie Universität Berlin

No degree, Mathematics, No degree, Mathematics at Westfälische Wilhelms-Universität Münster

MSc, Artificial Intelligence, cum laude, MSc, Artificial Intelligence, cum laude at University of Amsterdam

No Degree, Political Economics, No Degree, Political Economics at Ruprecht-Karls-Universität Heidelberg

No degree, Mathematics, No degree, Mathematics at Rheinische Friedrich-Wilhelms-Universität Bonn

Visiting Graduate Researcher, Mathematics, Visiting Graduate Researcher, Mathematics at University of California, Los Angeles

BSc, Computer Science and Media, distinction, BSc, Computer Science and Media, distinction at Bauhaus-Universität Weimar

German, English

Github Skills (6)

neural-network10

pytorch10

machine-learning10

convolutional-neural-networks10

python9

api-design7

Programming languages (2)

Jupyter NotebookPython

Github contributions (5)

jonkhler/s2cnn

Sep 2017 - Oct 2019

Spherical CNNs

Role in this project:

ML Engineer

Contributions:35 commits, 1 PR, 4 pushes in 2 years

Contributions summary:Jonas primarily focused on modifying and improving the example code within the repository, particularly related to the architecture and APIs of the spherical MNIST example. Their contributions included fixing API calls, updating the setup, and removing unused dependencies. The user was also involved in refactoring directory structures and making adjustments to the S2CNN's soft components.

cnnsmachine-learningspherical

jonas-koehler/ComputerGrafik

Oct 2015 - Jan 2016

Contributions:18 pushes, 1 branch, 1 tag in 2 months

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.

Request Free Trial