pytreez

An implementation of Jax pytrees in pure python (by shawwn)

Pytreez Alternatives

Similar projects and alternatives to pytreez

  • Pytorch

    341 pytreez VS Pytorch

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

  • penzai

    3 pytreez VS penzai

    A JAX research toolkit for building, editing, and visualizing neural networks.

  • 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
  • pytorch

    Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch (by Ascend)

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a better pytreez alternative or higher similarity.

pytreez reviews and mentions

Posts with mentions or reviews of pytreez. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2024-04-21.
  • penzai: JAX research toolkit for building, editing, and visualizing neural nets
    4 projects | news.ycombinator.com | 21 Apr 2024
    I implemented Jax’s pytrees in pure python. You can use it with whatever you want. https://github.com/shawwn/pytreez

    The readme is a todo, but the tests are complete. They’re the same that Jax itself uses, but zero dependencies. https://github.com/shawwn/pytreez/blob/master/tests/test_pyt...

    The concept is simple. The hard part is cross pollination. Suppose you wanted to literally use Jax pytrees with PyTorch. Now you’ll have to import Jax, or my library, and register your modules with it. But anything else that ever uses pytrees need to use the same pytree library, because the registry (the thing that keeps track of pytree compatible classes) is in the library you choose. They don’t share registries.

    A better way of phrasing it is that if you use a jax-style pytree interface, it should work with any other pytree library. But to my knowledge, the only pytree library besides Jax itself is mine here, and only I use it. So when you ask if pytree-compatible modules are compatible with PyTorch, it’s equivalent to asking whether PyTorch projects use jax, and the answer tends to be no.

Stats

Basic pytreez repo stats
1
1
-
over 2 years ago

The primary programming language of pytreez is Python.


Sponsored
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com