Career Profile

Machine Learning Engineer with extensive experience in the industry in fields such as autonomous vehicles, video analytics, and manufacturing. Contributor to various Python JAX-related open source projects (Flax, Elegy, Treex), speaker, host of the JAX Global Meetup and the Machine Learning Meetup Medellin.

Experiences

Software Enginner

2023 - Present
Google DeepMind

Lead developer of the Flax NNX Neural Networks API for JAX at DeepMind.

Senior Software Engineer (Machine Learning)

2020 - 2023
Quansight

Working on an Open Source Neural Networks framework with the Google Brain team.

Senior Machine Learning Engineer

2019 - 2020
Snappr Inc, Medellin

Developed applications using Machine Learning and Deep Learning to automate tasks throughout the organization such as client sales prediction, demand forecast, photo aesthetics prediction, image clustering, models using GIS data, and more.

Machine Learning Engineer

2018 - 2019
Landing AI, Medellin

Developed Automatic Visual Inspection solutions using Deep Learning techniques for clients in the automotive industry. Developed an internal tool to that enabled training deep learning models with small datasets. Worked on a generative model for images using GANs.

Lead Engineer in Computer Vision

6 months
Bigbang Media, CA

Developed various Deep Learning models for tasks like video classification and event detection in videos using distributed Tensorflow, TPUs, and Dataflow.

Lead Data Scientist

2016 - 2018
Kiwi Campus, Berkeley

Lead Data Scientist and Deep Learning developer for autonomous driving delivery robots. Used Deep Learning techniques like CNNs and RNNs for autonomous control, with technologies such as TensorFlow, ROS, Jupyter, Flask.

Data Scientist

2015 - 2016
BD Guidance, Medellin

Designed the curriculum of various Data Science/ML courses. Investigated technologies such a TensorFlow and OpenCV to run ML algorithms in IoT devices.

Data Scientist

2014 - 2015
Senseta, Medellin

Developed a Linux and Windows python application for HR software to measure employee productivity. Cleaned and analyzed large datasets using Spark and Zeppelin. Created a service in Python for Entity extracting using NLTK. Created a generic ML prediction service using Scikit Learning and Flask

Projects

Main developer of various Python Open Source projects and community leader.

Elegy - A High Level API for Deep Learning in JAX
Treex - A Pytree Module system for Deep Learning in JAX
Treeo - A small library for creating and manipulating custom JAX Pytree classes
Pypeln - Concurrent data pipelines in Python
Machine Learning Meetup Medellin - Organizer (3K+ members)
JAX Global Meetup - Organizer
Machine Learning Colombia Group - Admin (14K+ members)

Skills & Proficiency

Python

JAX

TensorFlow

Scikit-Learn

GCP

Docker

Kubeflow

FastAPI

AWS

Pytorch

Rust