extending-jax

Extending JAX with custom C++ and CUDA code (by dfm)
Review Jax Xla Cuda
Source Code
extending-jax Reviews
Suggest alternative
Edit details
InfluxDB – Built for High-Performance Time Series Workloads
InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.
www.influxdata.com
featured

Extending-jax Alternatives

Similar projects and alternatives to extending-jax

dfm
extending-jax

  1. jax

    90 extending-jax VS jax

    Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  2. Stream

    Stream - Scalable APIs for Chat, Feeds, Moderation, & Video. Stream helps developers build engaging apps that scale to millions with performant and flexible Chat, Feeds, Moderation, and Video APIs and SDKs powered by a global edge network and enterprise-grade infrastructure.

    Stream logo

  3. equinox

    33 extending-jax VS equinox

    Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

  4. diffrax

    23 extending-jax VS diffrax

    Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

  5. flax

    10 extending-jax VS flax

    Flax is a neural network library for JAX that is designed for flexibility.

  6. dm-haiku

    10 extending-jax VS dm-haiku

    JAX-based neural network library

  7. jax-models

    6 extending-jax VS jax-models

    Unofficial JAX implementations of deep learning research papers

  8. elegy

    5 extending-jax VS elegy

    A High Level API for Deep Learning in JAX

  9. InfluxDB

    InfluxDB – Built for High-Performance Time Series Workloads. InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.

    InfluxDB logo

  10. einops

    23 extending-jax VS einops

    Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

  11. thinc

    4 extending-jax VS thinc

    🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

  12. mpi4jax

    1 extending-jax VS mpi4jax

    Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

  13. trax

    7 extending-jax VS trax

    Trax — Deep Learning with Clear Code and Speed

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a better extending-jax alternative or higher similarity.
Suggest an alternative to extending-jax

extending-jax discussion

Log in or Post with

extending-jax reviews and mentions

Posts with mentions or reviews of extending-jax. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2022-02-15.

Stats

Basic extending-jax repo stats
2
398
3.0
11 months ago

dfm/extending-jax is an open source project licensed under MIT License which is an OSI approved license.

The primary programming language of extending-jax is Python.

Popular Comparisons

  1. extending-jax VS equinox
  2. extending-jax VS einops
  3. extending-jax VS thinc
  4. extending-jax VS mpi4jax
  5. extending-jax VS trax
  6. extending-jax VS elegy
  7. extending-jax VS jax-models
  8. extending-jax VS diffrax
  9. extending-jax VS flax
  10. extending-jax VS jax

Sponsored
Stream - Scalable APIs for Chat, Feeds, Moderation, & Video.
Stream helps developers build engaging apps that scale to millions with performant and flexible Chat, Feeds, Moderation, and Video APIs and SDKs powered by a global edge network and enterprise-grade infrastructure.
getstream.io

Did you know that Python is
the 2nd most popular programming language
based on number of references?

Loading...