Learn new Google tools with your community. Find a DevFest near you!
Stay organized with collections Save and categorize content based on your preferences.
Soumik Rakshit
Kolkata, West Bengal, India


Machine Learning (JAX/Flax)

My Biography

I am a Machine Learning Engineer working on end-to-end deep learning pipelines as part of the Growth team at Weights & Biases. My research and development interests include Generative Machine Learning Models, Video Enhancement Systems, Neural Radiance Fields, and NLP-based solutions to improve and automate student evaluation in the education sector. I enjoy implementing research papers in machine learning and computer graphics and communicating such ideas using end-to-end notebooks, and articles. I'm also working on training machine learning models on TPUs using Jax, Flax, Tensorflow, and Keras. More about me at github.com/soumik12345.