Avatar for The AI Foundation

Unlock what’s uniquely possible for you with your own ai that thinks and acts like you

Machine Learning Infrastructure Engineer

$130k – $180k • 0.1% – 0.25%
Apply now

AI Foundation is currently looking to add a software engineer to work closely with researchers to bring to production ideas across multiple disciplines: speech, natural language processing, and computer vision.

What this candidate will bring to the table:

  • Understanding of system level architecture, such as interconnects, memory hierarchy, interrupts, and memory-mapped IO
  • Professional experience writing performant scientific, numerical and parallel code in a data-driven research environment
  • Experience in performance profiling and characterization
  • Understand how to optimize things like cache locality and CPU core count vs clock speed
  • Experience building and maintaining software libraries
  • Good communication and organization skills, with a logical approach to problem-solving, good time management, and task prioritization skills

Specific skill sets this candidate will have:

  • Experience with deep learning frameworks: PyTorch (preferable), Tensorflow
  • Experience using Cython, Numba, C or similar to speed up analytical code
  • Experience writing and optimizing code in a vectorized scripting language (numpy or similar BLAS/LAPACK wrapper)
  • Strong knowledge of C/C++, Python, software design, programming techniques, and AI algorithms
  • Experience with GPU acceleration (CUDA and CUDNN)
  • Maintaining large scientific python codebases
  • Dealing with issues of providence and experimental repeatability
  • High-Performance Computing (HPC)

What we can offer:

This is a mid- or senior-level engineering role and we offer a competitive salary, valuable stock options and comprehensive benefits that allow individual flexibility for all employees to choose what makes the most sense for their personal situation.

AI Foundation is committed to a culture of flexibility, diversity, and fun for all of our employees. We are working on some of the most challenging technical problems and we know the solutions will come from all of us working together in an inclusive, transparent and open manner.

Meet your team

People you would work with in this role

Delip Rao

Avatar for Delip Rao
Researcher. I like to build stuff and teams. Previously at Amazon (Alexa), Google, Twitter. I like working at the intersection of science, people, and products.

Brent Haines

Avatar for Brent Haines
I am a technical leader who loves to develop and manage uncommonly effective software engineering teams.
Co-Founder @AIFoundation, Founder at @Originate

The AI Foundation at a glance

Unlock what’s uniquely possible for you with your own ai that thinks and acts like you

The AI Foundation focuses on Social Media, Artificial Intelligence, Software, Entertainment Industry, and Communications Software. Their company has offices in San Francisco and Las Vegas. They have a small team that's between 11-50 employees. To date, The AI Foundation has raised $10M of funding; their latest round was closed on September 2018.

You can view their website at http://www.aifoundation.com/ or find them on Twitter and LinkedIn.

More jobs at The AI Foundation

View all jobs

Research Scientist (NLP)

Sr. 3D Modeling Software Engineer / Visual Hacker

Research Scientist (Computer Vision)

Research Scientist (Speech Synthesis)

Unity 3D Developer

Technical Director - Visual Development

Similar jobs to Machine Learning Infrastructure Engineer at The AI Foundation