Productive, portable, and performant GPU programming in Python.
Swift for TensorFlow
10 differentiable physical simulators built with Taichi differentiable p...
A general-purpose probabilistic programming system with programmable inf...
Hardware accelerated, batchable and differentiable optimizers in JAX.
Swift for TensorFlow Deep Learning Library
Pytorch-based framework for solving parametric constrained optimization ...
A high-fidelity 3D face reconstruction library from monocular RGB image(s)
torchbearer: A model fitting library for PyTorch
🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types
TorchOpt is an efficient library for differentiable optimization built u...
⟨Grassmann-Clifford-Hodge⟩ multilinear differential geometric algebra
Code repository for our paper DiffCloth: Differentiable Cloth Simulatio...
Robot kinematics implemented in pytorch
[ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?