NVIDIA/warp

A Python framework for accelerated simulation, data generation and spatial computing.

Language: Python - Size: 48.5 MB - Last synced at: about 9 hours ago - Pushed at: about 11 hours ago - Stars: 5,242 - Forks: 326

hafuhauhfdhuidshui/WARP

WARP is a Rust-first toolkit for creating modular WebAssembly components, making it easier to build portable and composable WASM applications. With WIT interfaces and a focus on clear communication, developers can streamline their projects and enhance reusability. 🦀✨

Language: Rust - Size: 14.6 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 0 - Forks: 0

EnzymeAD/Enzyme

High-performance automatic differentiation of LLVM and MLIR.

Language: LLVM - Size: 628 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 1,415 - Forks: 131

pasteurlabs/tesseract-core

Autodiff-native, self-documenting software components for Simulation Intelligence. 📦

Language: Python - Size: 12.6 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 57 - Forks: 2

pnnl/neuromancer

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

Language: Python - Size: 408 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 1,137 - Forks: 150

ExtensityAI/symbolicai

Compositional Differentiable Programming Library

Language: Python - Size: 27.9 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 1,048 - Forks: 60

tensorflow/swift 📦

Swift for TensorFlow

Language: Jupyter Notebook - Size: 3.46 MB - Last synced at: 2 days ago - Pushed at: over 3 years ago - Stars: 6,138 - Forks: 610

acados/acados

Fast and embedded solvers for nonlinear optimal control and nonlinear model predictive control

Language: C - Size: 43.7 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 1,014 - Forks: 279

EnzymeAD/Enzyme.jl

Julia bindings for the Enzyme automatic differentiator

Language: Julia - Size: 7.71 MB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 504 - Forks: 71

XanaduAI/MrMustard

A differentiable bridge between phase space and Fock space

Language: Python - Size: 10.9 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 89 - Forks: 29

kach/memo

A domain-specific probabilistic programming language for reasoning about reasoning

Language: Python - Size: 42.4 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 78 - Forks: 6

control-toolbox/OptimalControl.jl

Model and solve optimal control problems in Julia

Language: Julia - Size: 112 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 105 - Forks: 9

pasteurlabs/tesseract-jax

Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡

Language: Python - Size: 10.3 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 15 - Forks: 0

tensorflow/swift-apis 📦

Swift for TensorFlow Deep Learning Library

Language: Swift - Size: 22.8 MB - Last synced at: 2 days ago - Pushed at: about 3 years ago - Stars: 794 - Forks: 137

genjax-community/genjax

Probabilistic programming with programmable inference for parallel accelerators.

Language: Python - Size: 53.8 MB - Last synced at: 5 days ago - Pushed at: 8 days ago - Stars: 24 - Forks: 5

taichi-dev/taichi

Productive, portable, and performant GPU programming in Python.

Language: C++ - Size: 57.4 MB - Last synced at: 9 days ago - Pushed at: about 1 month ago - Stars: 27,202 - Forks: 2,345

probcomp/Gen.jl

A general-purpose probabilistic programming system with programmable inference

Language: Julia - Size: 28.1 MB - Last synced at: 2 days ago - Pushed at: 13 days ago - Stars: 1,816 - Forks: 162

astro-informatics/s2wav

Differentiable and accelerated wavelet transform on the sphere with JAX

Language: Python - Size: 11.9 MB - Last synced at: 2 days ago - Pushed at: about 1 month ago - Stars: 20 - Forks: 0

SciML/NBodySimulator.jl

A differentiable simulator for scientific machine learning (SciML) with N-body problems, including astrophysical and molecular dynamics

Language: Julia - Size: 600 KB - Last synced at: 8 days ago - Pushed at: 9 months ago - Stars: 135 - Forks: 19

jump-dev/DiffOpt.jl

Differentiating convex optimization programs w.r.t. program parameters

Language: Julia - Size: 12.2 MB - Last synced at: 5 days ago - Pushed at: 7 days ago - Stars: 130 - Forks: 14

alonfnt/pcax

Differentiable Principal Component Analysis (PCA) implementation in JAX

Language: Python - Size: 35.2 KB - Last synced at: 5 days ago - Pushed at: 2 months ago - Stars: 28 - Forks: 1

astro-informatics/s2fft

S2FFT: Differentiable and accelerated spherical transforms

Language: Python - Size: 61.1 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 195 - Forks: 13

minliu01/SoftMAC

Code repository for our paper SoftMAC: Differentiable Soft Body Simulation with Forecast-based Contact Model and Two-way Coupling with Articulated Rigid Bodies and Clothes

Language: Python - Size: 64.2 MB - Last synced at: 21 days ago - Pushed at: 21 days ago - Stars: 30 - Forks: 4

OrchardLANL/DPFEHM.jl

DPFEHM: A Differentiable Subsurface Flow Simulator

Language: Julia - Size: 9.66 MB - Last synced at: 3 days ago - Pushed at: 9 months ago - Stars: 37 - Forks: 15

SciML/Integrals.jl

A common interface for quadrature and numerical integration for the SciML scientific machine learning organization

Language: Julia - Size: 2.17 MB - Last synced at: 4 days ago - Pushed at: about 1 month ago - Stars: 236 - Forks: 30

grinisrit/noa

Differentiable Programming Algorithms in Modern C++

Language: LLVM - Size: 153 MB - Last synced at: 24 days ago - Pushed at: 24 days ago - Stars: 150 - Forks: 29

sail-sg/jrystal

A JAX-based Differentiable Density Functional Theory Framework for Materials

Language: Python - Size: 6.07 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 15 - Forks: 2

arnauqb/blackbirds

Black-Box Inference foR Differentiable Simulators

Language: Python - Size: 15 MB - Last synced at: 3 days ago - Pushed at: 8 months ago - Stars: 20 - Forks: 1

LukasZahradnik/PyNeuraLogic

PyNeuraLogic lets you use Python to create Differentiable Logic Programs

Language: Python - Size: 31.9 MB - Last synced at: 4 days ago - Pushed at: about 2 months ago - Stars: 293 - Forks: 21

artificial-scientist-lab/Differometor

A Differentiable Interferometer Simulator for the Computational Design of Gravitational Wave Detectors

Language: Python - Size: 14.8 MB - Last synced at: 11 days ago - Pushed at: about 1 month ago - Stars: 3 - Forks: 0

VThuong99/differentiable-wonderland-solutions

Jupyter Notebook solutions for Alice’s Adventures in a Differentiable Wonderland using PyTorch.

Size: 10.7 KB - Last synced at: about 21 hours ago - Pushed at: about 1 month ago - Stars: 0 - Forks: 0

ucl-bug/jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

Language: Python - Size: 16.9 MB - Last synced at: 12 days ago - Pushed at: 10 months ago - Stars: 128 - Forks: 9

UM-ARM-Lab/pytorch_kinematics

Robot kinematics implemented in pytorch

Language: Python - Size: 1.28 MB - Last synced at: about 1 month ago - Pushed at: 3 months ago - Stars: 588 - Forks: 51

deepmodeling/jax-fem

Differentiable Finite Element Method with JAX

Language: Python - Size: 79.1 MB - Last synced at: about 1 month ago - Pushed at: about 2 months ago - Stars: 394 - Forks: 67

SciML/PreallocationTools.jl

Tools for building non-allocating pre-cached functions in Julia, allowing for GC-free usage of automatic differentiation in complex codes

Language: Julia - Size: 1.38 MB - Last synced at: 4 days ago - Pushed at: 10 days ago - Stars: 121 - Forks: 16

metaopt/torchopt

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

Language: Python - Size: 5.27 MB - Last synced at: about 1 month ago - Pushed at: about 2 months ago - Stars: 589 - Forks: 36

esa/dSGP4

dSGP4: differentiable SGP4. Supports differentiability, ML integration & embarassingly parallel computations

Language: Python - Size: 4.13 MB - Last synced at: 2 days ago - Pushed at: 2 months ago - Stars: 51 - Forks: 11

breandan/kotlingrad

🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types

Language: Kotlin - Size: 288 MB - Last synced at: about 1 month ago - Pushed at: 7 months ago - Stars: 537 - Forks: 22

eelregit/mcfit

multiplicatively convolutional fast integral transforms, implementing FFTLog

Language: Python - Size: 357 KB - Last synced at: 6 days ago - Pushed at: 4 months ago - Stars: 34 - Forks: 12

namin/relaxed-machines

program synthesis with neuro-symbolic differentiable interpreters

Language: Python - Size: 149 KB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 14 - Forks: 0

hypercoil/hypercoil

A differentiable program for mapping brain function

Language: Python - Size: 17 MB - Last synced at: 28 days ago - Pushed at: 8 months ago - Stars: 6 - Forks: 0

kach/gradient-descent-the-ultimate-optimizer

Code for our NeurIPS 2022 paper

Language: Python - Size: 5.86 KB - Last synced at: 4 days ago - Pushed at: over 2 years ago - Stars: 368 - Forks: 21

qiauil/ConvDO

Convolutional Differential Operators for Physics-based Deep Learning Study

Language: Python - Size: 7.14 MB - Last synced at: 3 days ago - Pushed at: 11 months ago - Stars: 25 - Forks: 5

emaballarin/foveatorch

👀🔥 Differentiable foveated vision for Deep Learning methods

Language: Python - Size: 1.51 MB - Last synced at: 13 days ago - Pushed at: 8 months ago - Stars: 6 - Forks: 0

rkruegs123/idp-design

Optimizing IDPs directly at the level of simulations via a continuous sequence representation.

Language: Jupyter Notebook - Size: 13.5 MB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 7 - Forks: 1

edxmorgan/diff_uv

A differentiable underwater vehicle dynamics in body and ned(euler & quaternion).

Language: Python - Size: 1.93 MB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 13 - Forks: 4

google/jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

Language: Python - Size: 3.35 MB - Last synced at: 2 months ago - Pushed at: 3 months ago - Stars: 960 - Forks: 70

bifurcationkit/BifurcationInference.jl

learning state-space targets in dynamical systems

Language: Julia - Size: 45.4 MB - Last synced at: 8 days ago - Pushed at: about 1 year ago - Stars: 28 - Forks: 2

ctongfei/nexus

Experimental tensor-typed deep learning

Language: Scala - Size: 3.37 MB - Last synced at: about 2 months ago - Pushed at: almost 6 years ago - Stars: 257 - Forks: 16

avik-pal/RayTracer.jl

Differentiable RayTracing in Julia

Language: Julia - Size: 34.9 MB - Last synced at: about 2 months ago - Pushed at: over 2 years ago - Stars: 152 - Forks: 12

pytorchbearer/torchbearer

torchbearer: A model fitting library for PyTorch

Language: Python - Size: 7.88 MB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 640 - Forks: 64

taichi-dev/difftaichi

10 differentiable physical simulators built with Taichi differentiable programming (DiffTaichi, ICLR 2020)

Size: 345 KB - Last synced at: 3 months ago - Pushed at: over 1 year ago - Stars: 2,558 - Forks: 261

Lucaman99/autohf

Automatically differentiable Hartree-Fock

Language: Python - Size: 507 KB - Last synced at: about 2 months ago - Pushed at: over 2 years ago - Stars: 1 - Forks: 0

matsunagalab/MDToolbox.jl

MDToolbox.jl: A Julia package for molecular dynamics trajectories analysis and modeling of biomolecules

Language: Julia - Size: 208 MB - Last synced at: 3 months ago - Pushed at: 3 months ago - Stars: 21 - Forks: 5

matsunagalab/differentiable

test-bed for differentiable parameter optimization of molecular systems

Language: MAXScript - Size: 736 MB - Last synced at: 3 months ago - Pushed at: 3 months ago - Stars: 0 - Forks: 0

matsunagalab/ColabBTR

End-to-end differentiable blind tip reconstruction on Colab implemented with PyTorch

Language: Jupyter Notebook - Size: 224 MB - Last synced at: about 2 months ago - Pushed at: 3 months ago - Stars: 2 - Forks: 3

kach/designing-perceptual-puzzles-by-differentiating-probabilistic-programs

Supplementary materials for our SIGGRAPH 2022 paper

Language: Jupyter Notebook - Size: 3.39 MB - Last synced at: 3 months ago - Pushed at: about 3 years ago - Stars: 28 - Forks: 3

cvxgrp/cocp

Source code for the examples accompanying the paper "Learning convex optimization control policies."

Language: Jupyter Notebook - Size: 3.43 MB - Last synced at: 3 months ago - Pushed at: over 2 years ago - Stars: 84 - Forks: 16

RePlasma/PhysRevA.103.052416

Solving nonlinear differential equations with differentiable quantum circuits

Language: Mathematica - Size: 187 KB - Last synced at: 4 months ago - Pushed at: 4 months ago - Stars: 0 - Forks: 0

SciCompKL/derivgrind

Automatic differentiation of compiled programs using the Valgrind framework.

Language: C - Size: 47.5 MB - Last synced at: 4 months ago - Pushed at: 4 months ago - Stars: 5 - Forks: 3

GustikS/NeuraLogic

Deep relational learning through differentiable logic programming.

Language: Java - Size: 3.65 MB - Last synced at: 3 months ago - Pushed at: 11 months ago - Stars: 102 - Forks: 15

infergo-ml/infergo

mirror of Infergo repository

Language: Go - Size: 791 KB - Last synced at: 4 months ago - Pushed at: 4 months ago - Stars: 10 - Forks: 1

chakravala/Grassmann.wl

⟨Leibniz-Grassmann-Clifford⟩ multilinear differential geometric algebra

Language: Mathematica - Size: 111 KB - Last synced at: 4 months ago - Pushed at: over 3 years ago - Stars: 6 - Forks: 3

junyixu/TrixiEnzyme.jl

Autodiff via Enzyme.jl for Trixi.jl

Language: Julia - Size: 600 KB - Last synced at: 3 days ago - Pushed at: 6 months ago - Stars: 2 - Forks: 1

alexjc/pytrain

🚃 Automated task/test framework for writing differentiable software.

Language: Python - Size: 243 KB - Last synced at: 3 months ago - Pushed at: over 4 years ago - Stars: 7 - Forks: 0

wangleiphy/TRG

Differentiable Tensor Renormalization Group for square Ising

Language: Julia - Size: 21.5 KB - Last synced at: 3 months ago - Pushed at: over 6 years ago - Stars: 17 - Forks: 4

divicomp/potto

A prototype differentiable programming language with an integration primitive that soundly handles interactions among the derivative, integral, and discontinuities and supports separate compilation.

Language: Python - Size: 370 KB - Last synced at: 3 months ago - Pushed at: 9 months ago - Stars: 5 - Forks: 0

belkakari/cellular-automata-pytorch

A reproduction and tweaking of Growing Neural Cellular Automata

Language: Python - Size: 13.1 MB - Last synced at: 2 months ago - Pushed at: over 5 years ago - Stars: 18 - Forks: 3

astro-informatics/s2scat

Differentiable and GPU accelerated scattering covariance statistics on the sphere

Language: Jupyter Notebook - Size: 281 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 10 - Forks: 2

astro-informatics/s2ball

Differentiable and accelerated wavelet transform on the ball with JAX

Language: Python - Size: 4.6 MB - Last synced at: 3 months ago - Pushed at: over 1 year ago - Stars: 6 - Forks: 0

OuyangWenyu/HydroDHM

A differentiable hydrological model for daily streamflow prediction in data-scarce basins

Language: Python - Size: 115 KB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 6 - Forks: 1

Andrew-Luo1/jax_shac

An implementation of Short Horizon Actor Critic writen in Jax. Core algorithm written in the style of Brax, with several bits taken from Xu's original paper.

Language: Jupyter Notebook - Size: 18.1 MB - Last synced at: 8 months ago - Pushed at: 8 months ago - Stars: 8 - Forks: 1

UTokyo-ICEPP/multiml

A prototype framework for developing multi-step machine learnings.

Language: Python - Size: 689 KB - Last synced at: 8 months ago - Pushed at: 8 months ago - Stars: 5 - Forks: 4

cesmix-mit/PotentialLearning.jl

PotentialLearning.jl: Optimize your atomistic data and interatomic potential models in your molecular dynamic workflows.

Language: Julia - Size: 116 MB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 21 - Forks: 10

ChezJrk/Teg

A differentiable programming language with an integration primitive that soundly handles interactions among the derivative, integral, and discontinuities.

Language: Python - Size: 257 KB - Last synced at: 3 months ago - Pushed at: about 1 year ago - Stars: 37 - Forks: 4

ewanwm/nuTens

Tensor based neutrino oscillation probability calculator

Language: C++ - Size: 848 KB - Last synced at: 5 days ago - Pushed at: 6 months ago - Stars: 0 - Forks: 0

TorchOPC/TorchLitho

Differentiable Computational Lithogrpahy Framework

Language: Python - Size: 5.46 MB - Last synced at: 10 months ago - Pushed at: 10 months ago - Stars: 156 - Forks: 31

matsunagalab/differentiable_BTR

Jupyter notebooks for end-to-end differentiable blind tip reconstruction

Language: Jupyter Notebook - Size: 315 MB - Last synced at: 3 months ago - Pushed at: 10 months ago - Stars: 0 - Forks: 3

ChezJrk/teg_applications

Applications of the Teg differentiable programming language to problems spanning graphics and physical simulation.

Language: Python - Size: 67.5 MB - Last synced at: 3 months ago - Pushed at: over 3 years ago - Stars: 11 - Forks: 2

FluxML/DiffImages.jl

Differentiable Computer Vision using pure Julia

Language: Julia - Size: 10.5 MB - Last synced at: 2 months ago - Pushed at: over 1 year ago - Stars: 15 - Forks: 2

a-n-onymous/schrl

Constrained Hierarchical Deep Reinforcement Learning with Differentiable Formal Specifications

Language: Python - Size: 5.71 MB - Last synced at: 6 days ago - Pushed at: over 2 years ago - Stars: 6 - Forks: 2

MatthiasNickles/diff-SAT

Probabilistic Answer Set Programming and Probabilistic SAT solving, based on Differentiable Satisfiability

Language: Scala - Size: 2.24 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 61 - Forks: 3

ZikangXiong/diff-spec

Differentiable Symbolic Specification

Language: Python - Size: 51.8 KB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 3 - Forks: 1

andrejobuljen/JaxPM-Hi-Fi-mocks Fork of DifferentiableUniverseInitiative/JaxPM

JAX-powered Hi-Fi mocks

Language: Jupyter Notebook - Size: 6.94 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 1 - Forks: 0

omegaiota/DiffCloth

Code repository for our paper DiffCloth: Differentiable Cloth Simulation with Dry Frictional Contact

Language: C++ - Size: 246 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 329 - Forks: 30

adam-r-kowalski/compute_graph

Language: Zig - Size: 1.07 MB - Last synced at: about 1 year ago - Pushed at: over 5 years ago - Stars: 3 - Forks: 0

LuisaGroup/luisa-compute-rs

Rust frontend to LuisaCompute and more!

Language: Rust - Size: 2.4 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 44 - Forks: 6

SciML/DiffEqDevMaterials

Various developer materials, like PDFs, notes, derivations, etc. for differential equations and scientific machine learning (SciML)

Language: Jupyter Notebook - Size: 680 KB - Last synced at: 9 days ago - Pushed at: about 1 year ago - Stars: 9 - Forks: 5

wangleiphy/hydrogen

variationally solve hydrogen atom with autograd

Language: Python - Size: 486 KB - Last synced at: over 1 year ago - Pushed at: about 6 years ago - Stars: 9 - Forks: 0

geoelements-dev/heat_transfer_duct_banks

Optimizing heat dissipation in duct banks using differentiable programming

Language: Python - Size: 213 KB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 1 - Forks: 1

wrsturgeon/dxpr

crates.io: Differentiable expression templates in Rust.

Language: Rust - Size: 46.9 KB - Last synced at: 14 days ago - Pushed at: over 2 years ago - Stars: 2 - Forks: 0

IvanYashchuk/jax-firedrake

Differentiable interface to Firedrake for JAX

Size: 3.91 KB - Last synced at: 4 months ago - Pushed at: over 4 years ago - Stars: 14 - Forks: 0

vnherdeiro/tensorflowjs-curve-fitting-on-graph

Language: TypeScript - Size: 4.9 MB - Last synced at: over 1 year ago - Pushed at: about 6 years ago - Stars: 2 - Forks: 0

DesmondZhong/diff_sim_improve_grads

PyTorch and Taichi implementations of our paper on improving gradient computation

Language: Python - Size: 298 KB - Last synced at: about 1 year ago - Pushed at: almost 2 years ago - Stars: 4 - Forks: 0

nbouziani/physics-driven-ml

Physics-driven machine learning using PyTorch and Firedrake

Language: Python - Size: 212 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 8 - Forks: 8

mlajtos/L1

L1: Tensor Studio — The playground for tensors

Language: JavaScript - Size: 12 MB - Last synced at: over 1 year ago - Pushed at: over 6 years ago - Stars: 184 - Forks: 25

Gsunshine/Enjoy-Hamburger

[ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?

Language: Python - Size: 4.36 MB - Last synced at: over 1 year ago - Pushed at: over 2 years ago - Stars: 306 - Forks: 20

abdallahdib/NextFace

A high-fidelity 3D face reconstruction library from monocular RGB image(s)

Language: Jupyter Notebook - Size: 14.5 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 599 - Forks: 83

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.

Language: Python - Size: 352 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 123 - Forks: 16

srikumarks/df

Code for understanding automatic differentiation.

Language: Haskell - Size: 17.6 KB - Last synced at: about 1 year ago - Pushed at: almost 3 years ago - Stars: 8 - Forks: 1