Stans Autodiff ist 4x schneller als JAX auf der CPU, aber 5x langsamer auf der GPU (in einer Evaluierung)
Das ist Bob. JAX im Kopf Ich habe viel darüber nachgedacht JAX in letzter Zeit. Für einen Informatiker wie mich ist JAX aufgrund seiner wunderbar kompositorischen Architektur zum Codieren von…