jax VS NATS

Compare jax vs NATS and see what are their differences.

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (by google)
Jax
Our great sponsors
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • WorkOS - The modern identity platform for B2B SaaS
  • SaaSHub - Software Alternatives and Reviews
jax NATS
82 106
27,936 14,720
4.0% 2.4%
10.0 9.8
1 day ago 7 days ago
Python Go
Apache License 2.0 Apache License 2.0
The number of mentions indicates the total number of mentions that we've tracked plus the number of user suggested alternatives.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.

jax

Posts with mentions or reviews of jax. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2024-03-22.
  • The Elements of Differentiable Programming
    5 projects | news.ycombinator.com | 22 Mar 2024
    The dual numbers exist just as surely as the real numbers and have been used well over 100 years

    https://en.m.wikipedia.org/wiki/Dual_number

    Pytorch has had them for many years.

    https://pytorch.org/docs/stable/generated/torch.autograd.for...

    JAX implements them and uses them exactly as stated in this thread.

    https://github.com/google/jax/discussions/10157#discussionco...

    As you so eloquently stated, "you shouldn't be proclaiming things you don't actually know on a public forum," and doubly so when your claimed "corrections" are so demonstrably and totally incorrect.

  • Julia GPU-based ODE solver 20x-100x faster than those in Jax and PyTorch
    6 projects | news.ycombinator.com | 23 Dec 2023
    On your last point, as long as you jit the topmost level, it doesn't matter whether or not you have inner jitted functions. The end result should be the same.

    Source: https://github.com/google/jax/discussions/5199#discussioncom...

  • Apple releases MLX for Apple Silicon
    4 projects | /r/LocalLLaMA | 8 Dec 2023
    The design of MLX is inspired by frameworks like NumPy, PyTorch, Jax, and ArrayFire.
  • MLPerf training tests put Nvidia ahead, Intel close, and Google well behind
    1 project | news.ycombinator.com | 14 Nov 2023
    I'm still not totally sure what the issue is. Jax uses program transformations to compile programs to run on a variety of hardware, for example, using XLA for TPUs. It can also run cuda ops for Nvidia gpus without issue: https://jax.readthedocs.io/en/latest/installation.html

    There is also support for custom cpp and cuda ops if that's what is needed: https://jax.readthedocs.io/en/latest/Custom_Operation_for_GP...

    I haven't worked with float4, but can imagine that new numerical types would require some special handling. But I assume that's the case for any ml environment.

    But really you probably mean fixed point 4bit integer types? Looks like that has had at least some work done in Jax: https://github.com/google/jax/issues/8566

  • MatX: Efficient C++17 GPU numerical computing library with Python-like syntax
    5 projects | news.ycombinator.com | 3 Oct 2023
    >

    Are they even comparing apples to apples to claim that they see these improvements over NumPy?

    > While the code complexity and length are roughly the same, the MatX version shows a 2100x over the Numpy version, and over 4x faster than the CuPy version on the same GPU.

    NumPy doesn't use GPU by default unless you use something like Jax [1] to compile NumPy code to run on GPUs. I think more honest comparison will mainly compare MatX running on same CPU like NumPy as focus the GPU comparison against CuPy.

    [1] https://github.com/google/jax

  • JAX – NumPy on the CPU, GPU, and TPU, with great automatic differentiation
    12 projects | news.ycombinator.com | 28 Sep 2023
    Actually that never changed. The README has always had an example of differentiating through native Python control flow:

    https://github.com/google/jax/commit/948a8db0adf233f333f3e5f...

    The constraints on control flow expressions come from jax.jit (because Python control flow can't be staged out) and jax.vmap (because we can't take multiple branches of Python control flow, which we might need to do for different batch elements). But autodiff of Python-native control flow works fine!

  • Julia and Mojo (Modular) Mandelbrot Benchmark
    10 projects | news.ycombinator.com | 8 Sep 2023
    For a similar "benchmark" (also Mandelbrot) but took place in Jax repo discussion: https://github.com/google/jax/discussions/11078#discussionco...
  • Functional Programming 1
    3 projects | news.ycombinator.com | 16 Aug 2023
    2. https://github.com/fantasyland/fantasy-land (A bit heavy on jargon)

    Note there is a python version of Ramda available on pypi and there’s a lot of FP tidbits inside JAX:

    3. https://pypi.org/project/ramda/ (Worth making your own version if you want to learn, though)

    4. For nested data, JAX tree_util is epic: https://jax.readthedocs.io/en/latest/jax.tree_util.html and also their curry implementation is funny: https://github.com/google/jax/blob/4ac2bdc2b1d71ec0010412a32...

    Anyway don’t put FP on a pedestal, main thing is to focus on the core principles of avoiding external mutation and making helper functions. Doesn’t always work because some languages like Rust don’t have legit support for currying (afaik in 2023 August), but in those cases you can hack it with builder methods to an extent.

    Finally, if you want to understand the middle of the midwit meme, check out this wiki article and connect the free monoid to the Kleene star (0 or more copies of your pattern) and Kleene plus (1 or more copies of your pattern). Those are also in regex so it can help you remember the regex symbols. https://en.wikipedia.org/wiki/Free_monoid?wprov=sfti1

    The simplest example might be {0}^* in which case

    0: “” // because we use *

  • Best Way to Learn JAX
    1 project | /r/learnmachinelearning | 13 May 2023
    Hello! I'm trying to learn JAX over the next couple of weeks. Ideally, I want to be comfortable with using it for projects after about 3 weeks to a month, although I understand that may not be realistic. I currently have experience with PyTorch and TensorFlow. How should I go about learning JAX? Is there a specific YouTube tutorial or online course I should use, or should I just use the tutorial on https://jax.readthedocs.io/? Any information, advice, or experience you can share would be much appreciated!
  • Codon: Python Compiler
    9 projects | news.ycombinator.com | 8 May 2023

NATS

Posts with mentions or reviews of NATS. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2024-04-04.
  • Implementing OTel Trace Context Propagation Through Message Brokers with Go
    4 projects | dev.to | 4 Apr 2024
    Several message brokers, such as NATS and database queues, are not supported by OpenTelemetry (OTel) SDKs. This article will guide you on how to use context propagation explicitly with these message queues.
  • NATS: First Impressions
    1 project | news.ycombinator.com | 3 Apr 2024
    https://nats.io/ (Tracker removed)

    > Connective Technology for Adaptive Edge & Distributed Systems

    > An Introduction to NATS - The first screencast

    I guess I don't need to know what it is

  • Interview with Sebastian Holstein, Founder of Qaze
    1 project | dev.to | 21 Mar 2024
    During our interview, we referred to NATS quite a few times! If you want to learn more about it, Sebastian suggests this tutorial series.
  • Sequential and parallel execution of long-running shell commands
    7 projects | news.ycombinator.com | 20 Mar 2024
    Pueue dumps the state of the queue to the disk as JSON every time the state changes, so when you have a lot of queued jobs this results in considerable disk io. I actually changed it to compress the state file via zstd which helped quite a bit but then eventually just moved on to running NATS [1] locally.

    [1] https://nats.io/

  • Revolutionizing Real-Time Alerts with AI, NATs and Streamlit
    6 projects | dev.to | 18 Feb 2024
    Imagine you have an AI-powered personal alerting chat assistant that interacts using up-to-date data. Whether it's a big move in the stock market that affects your investments, any significant change on your shared SharePoint documents, or discounts on Amazon you were waiting for, the application is designed to keep you informed and alert you about any significant changes based on the criteria you set in advance using your natural language. In this post, we will learn how to build a full-stack event-driven weather alert chat application in Python using pretty cool tools: Streamlit, NATS, and OpenAI. The app can collect real-time weather information, understand your criteria for alerts using AI, and deliver these alerts to the user interface.
  • New scalable, fault-tolerant, and efficient open-source MQTT broker
    8 projects | news.ycombinator.com | 18 Dec 2023
    Why wasn't NATS[1] used ?

    Written in Go, single-binary deployment... there's a lot to love about NATS !

    [1]https://nats.io/

  • Scripting with NATS.io support
    1 project | /r/devops | 30 Oct 2023
    require nats.io
  • Introducing “Database Performance at Scale”: A Free, Open Source Book
    4 projects | news.ycombinator.com | 5 Oct 2023
    About cost, see [1]. Also, S3 prices have been increasing and there's been a bunch of alternative offers for object store from other companies. I think people in here (HN) comment often about increasing costs of AWS offerings.

    Distributed systems and consensus are inherently hard problem, but there are a lot of implementations that you can study (like Etcd that you mention, or NATS [2], which I've been playing with and looks super cool so far :-p) if you want to understand the internals, on top of many books and papers released.

    Again, I never said it was "easy" to build distributed systems, I just don't think there's any esoteric knowledge to what S3 provides.

    --

    1: https://en.wikipedia.org/wiki/Economies_of_scale

    2: https://nats.io/

  • NATS: Connective Technology for Adaptive Edge and Distributed Systems
    1 project | news.ycombinator.com | 21 Sep 2023
  • Is it an antipattern to use the response channel as identifier
    1 project | /r/NATS_io | 31 Jul 2023
    I am in a project were nats.io is used. Someone thought, it would be a great idea to link data in an event with data in a response using the response channel name.

What are some alternatives?

When comparing jax and NATS you can also consider the following projects:

Numba - NumPy aware dynamic Python compiler using LLVM

RabbitMQ - Open source RabbitMQ: core server and tier 1 (built-in) plugins

functorch - functorch is JAX-like composable function transforms for PyTorch.

celery - Distributed Task Queue (development branch)

julia - The Julia Programming Language

redpanda - Redpanda is a streaming data platform for developers. Kafka API compatible. 10x faster. No ZooKeeper. No JVM!

Pytorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration

ZeroMQ - ZeroMQ core engine in C++, implements ZMTP/3.1

Cython - The most widely used Python to C compiler

Apache ActiveMQ - Mirror of Apache ActiveMQ

jax-windows-builder - A community supported Windows build for jax.

nsq - A realtime distributed messaging platform