taichi-dev/taichi
Productive, portable, and performant GPU programming in Python.
Topic: differentiable-programming
236 repositories
NVIDIA/warp
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
tensorflow/swift
Swift for TensorFlow
taichi-dev/difftaichi
10 differentiable physical simulators built with Taichi differentiable programming (DiffTaichi, ICLR 2020)
probcomp/Gen.jl
A general-purpose probabilistic programming system with programmable inference
EnzymeAD/Enzyme
High-performance automatic differentiation of LLVM and MLIR.
acados/acados
Fast and embedded solvers for nonlinear optimal control and nonlinear model predictive control
pnnl/neuromancer
Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.
google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
optiland/optiland
Comprehensive optical design, optimization, and analysis in Python, including GPU-accelerated and differentiable ray tracing via PyTorch.
abdallahdib/NextFace
A high-fidelity 3D face reconstruction library from monocular RGB image(s)
UM-ARM-Lab/pytorch_kinematics
Robot kinematics implemented in pytorch
tensorflow/swift-apis
Swift for TensorFlow Deep Learning Library
deepmodeling/jax-fem
Differentiable Finite Element Method with JAX
pytorchbearer/torchbearer
torchbearer: A model fitting library for PyTorch
metaopt/torchopt
TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
EnzymeAD/Enzyme.jl
Julia bindings for the Enzyme automatic differentiator
breandan/kotlingrad
🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types
omegaiota/DiffCloth
Code repository for our paper DiffCloth: Differentiable Cloth Simulation with Dry Frictional Contact
kach/gradient-descent-the-ultimate-optimizer
Code for our NeurIPS 2022 paper
Gsunshine/Enjoy-Hamburger
[ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?
LukasZahradnik/PyNeuraLogic
PyNeuraLogic lets you use Python to create Differentiable Logic Programs
ctongfei/nexus
Experimental tensor-typed deep learning
SciML/Integrals.jl
A common interface for quadrature and numerical integration for the SciML scientific machine learning organization
TorchOPC/TorchLitho
Differentiable Computational Lithogrpahy Framework
astro-informatics/s2fft
S2FFT: Differentiable and accelerated spherical transforms
wanxinjin/Pontryagin-Differentiable-Programming
A unified end-to-end learning and control framework that is able to learn a (neural) control objective function, dynamics equation, control policy, or/and optimal trajectory in a control system.
mlajtos/L1
L1: Tensor Studio — The playground for tensors
kach/memo
A probabilistic programming language for metacognitive modeling
grinisrit/noa
Differentiable Programming Algorithms in Modern C++