Roman Novak is a research scientist in deep learning with 7 years of experience bridging theory and production-grade ML systems, currently working on the science of deep learning at OpenAI after research roles at Google Brain and DeepMind. He has a strong mathematical background from École Polytechnique and ENS/Paris-Saclay and a track record of improving core ML tooling—contributing stability, numerical fixes and new features to JAX, Flax, TensorFlow Probability and the influential Neural Tangents library. Roman’s work spans probabilistic modelling, infinite-width neural network theory and practical engineering, evidenced by tasks added to BIG-bench and low-level fixes that reduce NaNs and rank-promotion bugs in XLA/JAX. Colleagues describe him as equally comfortable proving theorems in functional analysis and shipping robust, well-tested examples and layers used by the wider ML community.
Contributions:16 releases, 33 reviews, 446 commits in 3 years 8 months
Contributions summary:Roman updated tests and examples within the `google/neural-tangents` repository, focusing on compatibility with the new JAX optimizers API and related API updates. The changes included refactoring and cleaning up code, removing unused imports, and aligning the testing framework with changes in the underlying JAX library. The user also modified existing examples and added a new one, updating them to reflect the latest JAX optimizers API, and ensuring functionality and accuracy.
Flax is a neural network library for JAX that is designed for flexibility.
Role in this project:
ML Engineer
Contributions:8 reviews, 9 commits, 2 PRs in 5 months
Contributions summary:Roman primarily contributed to the `google/flax` repository, a neural network library for JAX, focusing on implementing and testing various convolutional layers and functionalities. Their work includes adding the `ConvLocal` layer, incorporating circular padding, and performing testing of convolution operations. They also made internal changes and refactored code to switch to subclassing in `flax.linen.Conv[Local]`.
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.