patrick-kidger / sympytorchView external linksLinks
Turning SymPy expressions into PyTorch modules.
☆156Jul 24, 2023Updated 2 years ago
Alternatives and similar repositories for sympytorch
Users that are interested in sympytorch are comparing it to the libraries listed below
Sorting:
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 4 years ago
- Convex potential flows☆85Nov 19, 2021Updated 4 years ago
- ☆11Apr 14, 2022Updated 3 years ago
- Turn SymPy expressions into trainable JAX expressions.☆360Feb 4, 2026Updated last week
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆91Oct 19, 2022Updated 3 years ago
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,553May 2, 2024Updated last year
- Probabilistic Numerics in Python.☆458Jul 3, 2025Updated 7 months ago
- ☆622Jan 23, 2026Updated 3 weeks ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,898Updated this week
- Manage a Julia project living within a Python package☆18Mar 8, 2024Updated last year
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,705Dec 30, 2024Updated last year
- Documentation:☆127May 22, 2023Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Oct 26, 2020Updated 5 years ago
- Rotation-Based Iterative Gaussianization algorithm.☆17Mar 18, 2023Updated 2 years ago
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆470Sep 4, 2025Updated 5 months ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆15Jan 18, 2019Updated 7 years ago
- Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)☆697Oct 22, 2022Updated 3 years ago
- ☆214Jan 31, 2022Updated 4 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,765Feb 4, 2026Updated last week
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆174Feb 11, 2022Updated 4 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,026Dec 17, 2025Updated last month
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- ☆22Dec 3, 2021Updated 4 years ago
- Official code for the ICLR 2021 paper Neural ODE Processes☆76May 1, 2022Updated 3 years ago
- A conda-smithy repository for julia.☆24Dec 12, 2025Updated 2 months ago
- A parallel ODE solver for PyTorch☆277Oct 3, 2024Updated last year
- Experiment Manager☆22Jan 15, 2020Updated 6 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Nov 5, 2021Updated 4 years ago
- ☆34Oct 5, 2020Updated 5 years ago
- MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).☆24Dec 16, 2023Updated 2 years ago
- A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.☆33Aug 27, 2020Updated 5 years ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Modern normalizing flows in Python. Simple to use and easily extensible.☆11Updated this week
- High-Performance Symbolic Regression in Python and Julia☆3,359Feb 5, 2026Updated last week
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆65Mar 20, 2024Updated last year
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Apr 1, 2022Updated 3 years ago
- Manifold-learning flows (ℳ-flows)☆233Nov 13, 2020Updated 5 years ago
- Code for the article "Accelerated Forward-Backward Optimization using Deep Learning"☆12Sep 15, 2021Updated 4 years ago
- SymPy based framework for optimized code generation for BSSN formulation of Einstein equation for heterogeneous platforms.☆11Aug 18, 2025Updated 5 months ago