Jax vs. Julia (Vs PyTorch)

This page summarizes the projects mentioned and recommended in the original post on news.ycombinator.com

Our great sponsors
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • WorkOS - The modern identity platform for B2B SaaS
  • SaaSHub - Software Alternatives and Reviews
  • Enzyme

    High-performance automatic differentiation of LLVM and MLIR. (by EnzymeAD)

  • Idk, Enzyme is pretty next gen, all the way down to LLVM code.

    https://github.com/EnzymeAD/Enzyme

  • functorch

    functorch is JAX-like composable function transforms for PyTorch.

  • Tangentially related but there is an effort to get some of the features of JAX into PyTorch: https://pytorch.org/functorch/

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo
  • Flux.jl

    Relax! Flux is the ML library that doesn't make you tensor

  • > In his item #1, he links to https://discourse.julialang.org/t/loaderror-when-using-inter... The issue is actually a Zygote bug, a Julia package for auto-differentiation, and is not directly related to Julia codebase (or Flux package) itself. Furthermore, the problematic code is working fine now, because DiffEqFlux has switched to Enzyme, which doesn't have that bug. He should first confirm whether the problem he is citing is actually a problem or not.

    > Item #2, again another Zygote bug.

    If flux chose a buggy package as a dependency, that's on them, and users are well justified in steering clear of Flux if it has buggy dependencies. As of today, the Project.toml for both Flux and DiffEqFlux still lists Zygote as a dependency. Neither list Enzyme.

    https://github.com/FluxML/Flux.jl/blob/master/Project.toml

  • DiffEqFlux.jl

    Pre-built implicit layer architectures with O(1) backprop, GPUs, and stiff+non-stiff DE solvers, demonstrating scientific machine learning (SciML) and physics-informed machine learning methods

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts