Jax-md Alternatives

Similar projects and alternatives to jax-md

• jax

  82 27,936 10.0 Python jax-md VS jax

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

  

• tinygrad

  58 17,800 9.7 Python jax-md VS tinygrad

  Discontinued You like pytorch? You like micrograd? You love tinygrad! ❤️ [Moved to: https://github.com/tinygrad/tinygrad] (by geohot)

  

• InfluxDB

  www.influxdata.com sponsored

  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.

  

• equinox

  31 1,809 9.2 Python jax-md VS equinox

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

• diffrax

  21 1,230 8.3 Python jax-md VS diffrax

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

• torchmd

  1 507 7.9 Python jax-md VS torchmd

  End-To-End Molecular Dynamics (MD) Engine using PyTorch

  

• thinc

  4 2,787 6.9 Python jax-md VS thinc

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

  

• jaxonnxruntime

  1 70 8.3 Python jax-md VS jaxonnxruntime

  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.

  

• WorkOS

  workos.com sponsored

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

• autograd

  6 6,787 6.0 Python jax-md VS autograd

  Efficiently computes derivatives of numpy code.

  

• jax-experiments

  1 2 3.5 Jupyter Notebook jax-md VS jax-experiments

• faster-cpython

  20 931 0.0 jax-md VS faster-cpython

  How to make CPython faster.