Record · corpus.blog/posts/01a08795-8b3f-7127-9726-0507f1854150

JAX vs Julia (vs PyTorch)

kidger.site · published 3 May 2022

A record is what was published and who pointed at it. The text of the post is not held here: read it at the source, or at its Wayback capture.

Read the post at kidger.site Wayback capture

1blogs citing

First cited: 28 Dec 2014
Most recent: 28 Dec 2014
Rank this month: not ranked
Links from this post: 18
Words captured: 1,803

Cited by

Blog

In the post

Date

Dan Luu

A review of the Julia language Read at the source ↗

28 Dec 2014

Links from this post

Link

Host

https://discourse.julialang.org/t/state-of-machine-learning-in-julia/74385/4

discourse.julialang.org

https://github.com/google/jax

github.com

https://kidger.site/thoughts/https://kidger.site/thoughts/jax-vs-julia/julialang.org

kidger.site

https://github.com/patrick-kidger/equinox

github.com

https://github.com/FluxML/Flux.jl

github.com

https://horace.io/brrr_intro.html

horace.io

https://en.wikipedia.org/wiki/Homoiconicity

en.wikipedia.org

https://jax.readthedocs.io/en/latest/jaxpr.html

jax.readthedocs.io

https://docs.julialang.org/en/v1/manual/metaprogramming/

docs.julialang.org

https://docs.julialang.org/en/v1.8-dev/NEWS/

docs.julialang.org

https://github.com/google/jax/issues/9132

github.com

https://github.com/patrick-kidger/diffrax/blob/15c2ab9145ab0eda69af22d98d27f8ed02c90977/diffrax/integrate.py

github.com

https://discourse.julialang.org/t/loaderror-when-using-interpolations-as-input-for-a-neural-ode/51224/6

discourse.julialang.org

https://jax.readthedocs.io/

jax.readthedocs.io

https://pytorch.org/docs/stable/index.html

pytorch.org

https://jax.readthedocs.io/en/latest/autodidax.html

jax.readthedocs.io

https://github.com/patrick-kidger/diffrax

github.com

https://www.reddit.com/r/MachineLearning/comments/u34oh2/d_what_jax_nn_library_to_use/

reddit.com