Extending-jax Alternatives

Similar projects and alternatives to extending-jax

• jax

  82 28,082 10.0 Python extending-jax VS jax

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

  

• equinox

  31 1,827 9.2 Python extending-jax VS equinox

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

• InfluxDB

  www.influxdata.com featured

  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.

  

• diffrax

  21 1,249 8.2 Python extending-jax VS diffrax

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

• flax

  10 5,538 9.7 Python extending-jax VS flax

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

  

• dm-haiku

  10 2,813 7.8 Python extending-jax VS dm-haiku

  JAX-based neural network library

  

• elegy

  5 463 0.0 Python extending-jax VS elegy

  A High Level API for Deep Learning in JAX

  

• einops

  19 7,942 7.4 Python extending-jax VS einops

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

• SaaSHub

  www.saashub.com featured

  SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

  

• mpi4jax

  1 373 6.7 Python extending-jax VS mpi4jax

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

  

• jax-models

  6 138 0.0 Python extending-jax VS jax-models

  Unofficial JAX implementations of deep learning research papers

• thinc

  4 2,796 7.6 Python extending-jax VS thinc

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

  

• trax

  7 7,960 4.7 Python extending-jax VS trax

  Trax — Deep Learning with Clear Code and Speed