Turning SymPy expressions into PyTorch modules.
☆157Jul 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:
- Initial Commit☆19Dec 1, 2020Updated 5 years ago
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 4 years ago
- Convex potential flows☆85Nov 19, 2021Updated 4 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Mar 30, 2021Updated 4 years ago
- ☆11Apr 14, 2022Updated 3 years ago
- Turn SymPy expressions into trainable JAX expressions.☆363Feb 4, 2026Updated last month
- Probabilistic Solution of Differential Equations☆13Jun 19, 2022Updated 3 years ago
- 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
- ☆624Jan 23, 2026Updated last month
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,919Feb 23, 2026Updated last 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,704Dec 30, 2024Updated last year
- Documentation:☆129May 22, 2023Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Oct 26, 2020Updated 5 years ago
- lecture materials of the ML for Physics course 2021 in Perimeter Institute☆21Mar 31, 2021Updated 4 years ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- Rotation-Based Iterative Gaussianization algorithm.☆17Mar 18, 2023Updated 2 years ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆122May 11, 2021Updated 4 years ago
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆473Sep 4, 2025Updated 6 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)☆699Oct 22, 2022Updated 3 years ago
- ☆213Jan 31, 2022Updated 4 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,803Updated this 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,028Dec 17, 2025Updated 2 months ago
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Official code for the ICLR 2021 paper Neural ODE Processes☆76May 1, 2022Updated 3 years ago
- A conda-smithy repository for julia.☆24Feb 13, 2026Updated 3 weeks ago
- Experiment Manager☆22Jan 15, 2020Updated 6 years ago
- A parallel ODE solver for PyTorch☆276Oct 3, 2024Updated last year
- ☆22Dec 3, 2021Updated 4 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Nov 5, 2021Updated 4 years ago
- ☆34Oct 5, 2020Updated 5 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
- MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).☆25Dec 16, 2023Updated 2 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆20Feb 14, 2024Updated 2 years ago
- Modern normalizing flows in Python. Simple to use and easily extensible.☆12Feb 11, 2026Updated 3 weeks ago