diffrax

Machine-Learning Open Source

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

★ 2,108 Stars ⑂ 191 Forks License: Apache License 2.0

Download Latest Release Official Website ↗ GitHub Repository ↗

About diffrax

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

Supported Platforms

deep-learning differential-equations dynamical-systems equinox jax machine-learning neural-differential-equations neural-networks

Related in Machine-Learning

tensorflow

An Open Source Machine Learning Framework for Everyone

prompts.chat

f.k.a. Awesome ChatGPT Prompts. Share, discover, and collect prompts from the community. Free and open source — self-host for your organization with complete privacy.

transformers

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

LLMs-from-scratch

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

ML-For-Beginners

12 weeks, 26 lessons, 52 quizzes, classic Machine Learning for all