rtqichen / torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
☆5,471Updated 11 months ago
Related projects: ⓘ
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,526Updated this week
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,360Updated 4 months ago
- Fast and Easy Infinite Neural Networks in Python☆2,267Updated 6 months ago
- A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical…☆1,304Updated this week
- Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations☆669Updated 6 months ago
- Bayesian optimization in PyTorch☆3,060Updated this week
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,550Updated 3 months ago
- A Python toolbox for performing gradient-free optimization☆3,931Updated this week
- Efficiently computes derivatives of NumPy code.☆6,940Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,362Updated this week
- Code for visualizing the loss landscape of neural nets☆2,770Updated 2 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,012Updated 5 months ago
- A small package to create visualizations of PyTorch execution graphs☆3,168Updated 5 months ago
- JAX-based neural network library☆2,860Updated last week
- Flax is a neural network library for JAX that is designed for flexibility.☆5,950Updated this week
- Differentiable convex optimization layers☆1,785Updated last month
- Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)☆2,568Updated last year
- Build Graph Nets in Tensorflow☆5,344Updated last year
- AI conference deadline countdowns☆5,582Updated this week
- Hopfield Networks is All You Need☆1,660Updated last year
- A scikit-learn compatible neural network library that wraps PyTorch☆5,801Updated last week
- Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more☆1,817Updated 10 months ago
- Model interpretability and understanding for PyTorch☆4,810Updated this week
- POT : Python Optimal Transport☆2,382Updated last week
- Experiments with Deep Learning☆1,346Updated last year
- arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv☆5,193Updated last month
- A Collection of Variational Autoencoders (VAE) in PyTorch.☆6,467Updated 3 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,512Updated this week
- Awesome resources on normalizing flows.☆1,389Updated this week
- A Python implementation of global optimization with gaussian processes.☆7,806Updated last week