jax-md
torchmd
jax-md | torchmd | |
---|---|---|
2 | 1 | |
1,093 | 509 | |
- | 1.6% | |
7.5 | 7.9 | |
17 days ago | 2 months ago | |
Jupyter Notebook | Python | |
Apache License 2.0 | MIT License |
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-md
- JAX – NumPy on the CPU, GPU, and TPU, with great automatic differentiation
-
PyTorch 2.0
On the other hand, there is just no MD implemented with PyTorch.
[1]: https://github.com/jax-md/jax-md
torchmd
What are some alternatives?
jaxonnxruntime - A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
lightning-flash - Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
tinygrad - You like pytorch? You like micrograd? You love tinygrad! ❤️ [Moved to: https://github.com/tinygrad/tinygrad]
traingenerator - 🧙 A web app to generate template code for machine learning
jax-experiments
pytorch-forecasting - Time series forecasting with PyTorch
thinc - 🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
diffrax - Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
quaterion - Blazing fast framework for fine-tuning similarity learning models
equinox - Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/