Fetching the paper…

Reading the bibliography…

2018

Automatic Full Compilation of Julia Programs and ML Models to Cloud TPUs

Fischer, Keno, Saba, Elliot

Original
PDF
Post on X

Understand

Google's Cloud TPUs are a promising new hardware architecture for machine learning workloads.

They have powered many of Google's milestone machine learning achievements in recent years.
Google has now made TPUs available for general use on their cloud platform and as of very recently has opened them up further to allow use by non-TensorFlow frontends.
We describe a method and implementation for offloading suitable sections of Julia programs to TPUs via this new API and the Google XLA compiler.

Automatic Full Compilation of Julia Programs and ML Models to Cloud TPUs · Around

Built on

Llvm: A compilation framework for lifelong program analysis & transformation
Lattner, C. and Adve, V · 2004
Earlier work this paper cites.
Very deep convolutional networks for large-scale image recognition
Original
Simonyan, K. and Zisserman, A · 2014
Earlier work this paper cites.
Tensorflow: a system for large-scale machine learning
Abadi, M., Barham, P., Chen, J., Chen, Z., Davis, A., Dean, J., Devin, M., Ghemawat, S., Irving, G., Isard, M., et al · 2016
Earlier work this paper cites.
Flux.jl: The ml library that doesn’t make you tensor, 2017
Innes, M. and Contributors · 2017
Earlier work this paper cites.
More dots: Syntactic loop fusion in julia, 2017
Johnson, S. G · 2017
Earlier work this paper cites.

Similar

Differentialequations. jl–a performant and feature-rich ecosystem for solving differential equations in julia
Rackauckas, C. and Nie, Q · 2017
Cited alongside, same era.
Staticarrays.jl: Statically sized arrays for julia, 2018
Ferris, A. and Contributors · 2018
Cited alongside, same era.
Compiling machine learning programs via high-level tracing
Frostig, R., Johnson, M. J., and Leary, C · 2018
Cited alongside, same era.
Don’t Unroll Adjoint: Differentiating SSA-Form Programs
Innes, M · 2018
Cited alongside, same era.

Then

Structsofarrays.jl: Structures of arrays that behave like arrays of structures, 2018
Kornblith, S. and Contributors · 2018
Closest in time.
Metalhead.jl: Computer vision models for flux, 2018
Mike Innes, A. P. and Contributors · 2018
Closest in time.
Julia subtyping reconstructed, 2018
Nardelli, F. Z., Pelenitsyn, A., Chung, B., Vitek, J., and Bezanson, J · 2018
Closest in time.
Cassette.jl: Overdub your julia code, 2018
Revels, J. and Contributors · 2018
Closest in time.

Beyond the bibliography

alphaXiv searches the wider corpus for related work and actual follow-ups.

Open on alphaXiv

alphaXiv is searching for related work…