acados/acados

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

Language: C - Size: 49 MB - Last synced at: about 22 hours ago - Pushed at: about 24 hours ago - Stars: 1,140 - Forks: 305

EnzymeAD/Enzyme

High-performance automatic differentiation of LLVM and MLIR.

Language: LLVM - Size: 629 MB - Last synced at: about 16 hours ago - Pushed at: about 16 hours ago - Stars: 1,481 - Forks: 144

NVIDIA/warp

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

Language: Python - Size: 51.1 MB - Last synced at: 1 day ago - Pushed at: 2 days ago - Stars: 5,724 - Forks: 376

madgraph-ml/madevent7

Modular and GPU-ready phase-space library

Language: C++ - Size: 798 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 0 - Forks: 0

kach/memo

A probabilistic programming language for metacognitive modeling

Language: Python - Size: 42.3 MB - Last synced at: 1 day ago - Pushed at: 2 days ago - Stars: 123 - Forks: 8

google/jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

Language: Python - Size: 3.35 MB - Last synced at: 3 days ago - Pushed at: 23 days ago - Stars: 1,001 - Forks: 72

EnzymeAD/Enzyme.jl

Julia bindings for the Enzyme automatic differentiator

Language: Julia - Size: 11.8 MB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 524 - Forks: 79

arnauqb/blackbirds

Black-Box Inference foR Differentiable Simulators

Language: Python - Size: 15 MB - Last synced at: 1 day ago - Pushed at: 12 months ago - Stars: 21 - Forks: 1

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: 40.6 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 23 - Forks: 1

control-toolbox/OptimalControl.jl

Model and solve optimal control problems in Julia, both on CPU and GPU.

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

tensorflow/swift 📦

Swift for TensorFlow

Language: Jupyter Notebook - Size: 3.46 MB - Last synced at: 4 days ago - Pushed at: almost 4 years ago - Stars: 6,138 - Forks: 612

jump-dev/DiffOpt.jl

Differentiating optimization programs w.r.t. program parameters

Language: Julia - Size: 19 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 132 - Forks: 15

jeertmans/fpt-jax

Fermat path tracing with JAX

Language: Python - Size: 65.4 KB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 1 - Forks: 0

OuyangWenyu/HydroDHM

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

Language: Python - Size: 461 KB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 12 - Forks: 3

probcomp/Gen.jl

A general-purpose probabilistic programming system with programmable inference

Language: Julia - Size: 37.3 MB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 1,822 - Forks: 162

astro-informatics/s2fft

S2FFT: Differentiable and accelerated spherical transforms

Language: Python - Size: 75.2 MB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 208 - Forks: 13

pasteurlabs/tesseract-core

Universal, autodiff-native software components for Simulation Intelligence. 📦

Language: Python - Size: 15.2 MB - Last synced at: 6 days ago - Pushed at: 12 days ago - Stars: 73 - Forks: 4

UM-ARM-Lab/pytorch_kinematics

Robot kinematics implemented in pytorch

Language: Python - Size: 1.29 MB - Last synced at: 14 days ago - Pushed at: 15 days ago - Stars: 676 - Forks: 57

SciML/Integrals.jl

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

Language: Julia - Size: 2.47 MB - Last synced at: 14 days ago - Pushed at: 15 days ago - Stars: 239 - Forks: 33

tensorflow/swift-apis 📦

Swift for TensorFlow Deep Learning Library

Language: Swift - Size: 22.8 MB - Last synced at: 4 days ago - Pushed at: over 3 years ago - Stars: 791 - Forks: 138

pnnl/neuromancer

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

Language: Python - Size: 413 MB - Last synced at: 13 days ago - Pushed at: 15 days ago - Stars: 1,205 - Forks: 163

genjax-community/genjax

Probabilistic programming with programmable inference for parallel accelerators.

Language: Python - Size: 53.8 MB - Last synced at: 9 days ago - Pushed at: 13 days ago - Stars: 36 - Forks: 6

debangshu-mukherjee/janssen

Differentiable and Modular Optical Microscopy Simulations and Inversions

Language: Jupyter Notebook - Size: 8.55 MB - Last synced at: 18 days ago - Pushed at: 18 days ago - Stars: 1 - Forks: 0

matsunagalab/differentiable

test-bed for differentiable parameter optimization of molecular systems

Language: MAXScript - Size: 737 MB - Last synced at: 25 days ago - Pushed at: 25 days ago - Stars: 0 - Forks: 0

taichi-dev/taichi

Productive, portable, and performant GPU programming in Python.

Language: C++ - Size: 57.5 MB - Last synced at: 26 days ago - Pushed at: 26 days ago - Stars: 27,581 - Forks: 2,363

SciML/NBodySimulator.jl

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

Language: Julia - Size: 621 KB - Last synced at: 11 days ago - Pushed at: 3 months ago - Stars: 137 - Forks: 19

pytorchbearer/torchbearer

torchbearer: A model fitting library for PyTorch

Language: Python - Size: 7.88 MB - Last synced at: 10 days ago - Pushed at: almost 2 years ago - Stars: 641 - Forks: 64

metaopt/torchopt

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

Language: Python - Size: 5.26 MB - Last synced at: 17 days ago - Pushed at: 26 days ago - Stars: 616 - Forks: 39

esa/dSGP4

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

Language: Python - Size: 4.13 MB - Last synced at: 27 days ago - Pushed at: 7 months ago - Stars: 66 - Forks: 12

deepmodeling/jax-fem

Differentiable Finite Element Method with JAX

Language: Python - Size: 86.3 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 493 - Forks: 83

sail-sg/jrystal

A JAX-based Differentiable Density Functional Theory Framework for Materials

Language: Python - Size: 7.61 MB - Last synced at: 26 days ago - Pushed at: 26 days ago - Stars: 35 - Forks: 2

SuperkakaSCU/JAX-PF

Efficient GPU-computing simulation platform for differentiable phase field model

Language: Python - Size: 9.77 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 7 - Forks: 1

grinisrit/noa

Differentiable Programming Algorithms in Modern C++

Language: LLVM - Size: 154 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 155 - Forks: 31

namin/relaxed-machines

program synthesis with neuro-symbolic differentiable interpreters

Language: Python - Size: 157 KB - Last synced at: 22 days ago - Pushed at: about 1 month ago - Stars: 14 - Forks: 0

XanaduAI/MrMustard

A differentiable bridge between phase space and Fock space

Language: Python - Size: 9.7 MB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 90 - Forks: 29

astro-informatics/s2wav

Differentiable and accelerated wavelet transform on the sphere with JAX

Language: Python - Size: 11.9 MB - Last synced at: about 2 months ago - Pushed at: 5 months ago - Stars: 23 - Forks: 0

OrchardLANL/DPFEHM.jl

DPFEHM: A Differentiable Subsurface Flow Simulator

Language: Julia - Size: 9.75 MB - Last synced at: 3 days ago - Pushed at: 4 months ago - Stars: 40 - Forks: 14

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.3 MB - Last synced at: 11 days ago - Pushed at: about 2 months ago - Stars: 123 - Forks: 21

TorchOPC/TorchLitho

Differentiable Computational Lithogrpahy Framework

Language: Python - Size: 5.31 MB - Last synced at: about 2 months ago - Pushed at: 2 months ago - Stars: 192 - Forks: 36

ewanwm/nuTens

Tensor based engine for calculating neutrino oscillation probabilities in a fast, flexible, and differentiable way

Language: C++ - Size: 7.02 MB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 0 - Forks: 0

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: about 2 months ago - Pushed at: almost 3 years ago - Stars: 85 - Forks: 16

astro-informatics/s2scat

Differentiable and GPU accelerated scattering covariance statistics on the sphere

Language: Jupyter Notebook - Size: 281 MB - Last synced at: about 2 months ago - Pushed at: 5 months ago - Stars: 13 - Forks: 2

ctongfei/nexus

Experimental tensor-typed deep learning

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

artificial-scientist-lab/Differometor

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

Language: Python - Size: 14.9 MB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 7 - Forks: 1

Ghost---Shadow/sudoku-backprop-nmr

Solving SuDoKu with backprop and take an NMR of it while at it

Language: Python - Size: 1.94 MB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 0 - Forks: 0

ALFA-group/adversarial-code-generation

[ICLR 2021] "Generating Adversarial Computer Programs using Optimized Obfuscations" by Shashank Srikant, Sijia Liu, Tamara Mitrovska, Shiyu Chang, Quanfu Fan, Gaoyuan Zhang, and Una-May O'Reilly

Language: Python - Size: 16.2 MB - Last synced at: about 2 months ago - Pushed at: almost 4 years ago - Stars: 29 - Forks: 5

matsunagalab/ColabBTR

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

Language: Jupyter Notebook - Size: 67 MB - Last synced at: 12 days ago - Pushed at: 2 months ago - Stars: 4 - Forks: 3

oEmanuelFirmino/neuro-sym-model

A Python-based neuro-symbolic AI framework combining neural networks and symbolic logic. Implements custom tensors with autograd, neural layers, logic ASTs, and an interpreter to learn from facts and rules. Supports training, inference, and command-line execution without deep learning libraries.

Language: Python - Size: 122 KB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 1 - Forks: 0

breandan/kotlingrad

🧩 Shape-Safe Symbolic Differentiation with Algebraic Data Types

Language: Kotlin - Size: 288 MB - Last synced at: about 2 months ago - Pushed at: 11 months ago - Stars: 539 - Forks: 22

alonfnt/cinx

Minimal, differentible, fast cubic interpolation in JAX

Language: Python - Size: 8.79 KB - Last synced at: 2 days ago - Pushed at: 4 months ago - Stars: 2 - Forks: 0

kach/gradient-descent-the-ultimate-optimizer

Code for our NeurIPS 2022 paper

Language: Python - Size: 5.86 KB - Last synced at: 6 days ago - Pushed at: almost 3 years ago - Stars: 369 - Forks: 21

rkruegs123/idp-design

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

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

GustikS/NeuraLogic

Deep relational learning through differentiable logic programming.

Language: Java - Size: 3.67 MB - Last synced at: 3 months ago - Pushed at: 3 months ago - Stars: 106 - Forks: 15

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: 2 months ago - Pushed at: about 1 year ago - Stars: 130 - Forks: 10

ml4fp/2025-lbnl

ML4FP 2025: notebooks used for the Machine Learning for Fundamental Physics (ML4FP) School 2025

Language: Jupyter Notebook - Size: 54.8 MB - Last synced at: 3 months ago - Pushed at: 3 months ago - Stars: 16 - Forks: 5

taichi-dev/difftaichi

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

Size: 345 KB - Last synced at: 4 months ago - Pushed at: about 2 years ago - Stars: 2,627 - Forks: 268

alonfnt/pcax

Differentiable Principal Component Analysis (PCA) implementation in JAX

Language: Python - Size: 35.2 KB - Last synced at: about 1 month ago - Pushed at: 7 months ago - Stars: 29 - Forks: 1

LukasZahradnik/PyNeuraLogic

PyNeuraLogic lets you use Python to create Differentiable Logic Programs

Language: Python - Size: 35.6 MB - Last synced at: about 1 month ago - Pushed at: about 2 months ago - Stars: 294 - Forks: 22

hypercoil/hypercoil

A differentiable program for mapping brain function

Language: Python - Size: 17 MB - Last synced at: about 1 month ago - Pushed at: about 1 year ago - Stars: 7 - Forks: 0

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: 5 months ago - Pushed at: 5 months ago - Stars: 30 - Forks: 4

LuisaGroup/luisa-compute-rs

Rust frontend to LuisaCompute and more!

Language: Rust - Size: 2.42 MB - Last synced at: 2 months ago - Pushed at: 11 months ago - Stars: 56 - Forks: 9

VThuong99/differentiable-wonderland-solutions

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

Size: 10.7 KB - Last synced at: 4 months ago - Pushed at: 5 months ago - Stars: 0 - Forks: 0

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: 154 - Forks: 12

eelregit/mcfit

multiplicatively convolutional fast integral transforms, implementing FFTLog

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

qiauil/ConvDO

Convolutional Differential Operators for Physics-based Deep Learning Study

Language: Python - Size: 7.14 MB - Last synced at: about 1 month ago - Pushed at: over 1 year ago - Stars: 25 - Forks: 5

emaballarin/foveatorch

👀🔥 Differentiable foveated vision for Deep Learning methods

Language: Python - Size: 1.51 MB - Last synced at: 2 months ago - Pushed at: 12 months ago - Stars: 6 - Forks: 0

edxmorgan/diff_uv

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

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

bifurcationkit/BifurcationInference.jl

learning state-space targets in dynamical systems

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

Lucaman99/autohf

Automatically differentiable Hartree-Fock

Language: Python - Size: 507 KB - Last synced at: 6 months ago - Pushed at: almost 3 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: 8 months ago - Stars: 21 - Forks: 5

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: 7 months ago - Pushed at: over 3 years ago - Stars: 28 - Forks: 3

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: 416 KB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 5 - Forks: 0

RePlasma/PhysRevA.103.052416

Solving nonlinear differential equations with differentiable quantum circuits

Language: Mathematica - Size: 187 KB - Last synced at: 8 months ago - Pushed at: 8 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: 8 months ago - Pushed at: 8 months ago - Stars: 5 - Forks: 3

infergo-ml/infergo

mirror of Infergo repository

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

chakravala/Grassmann.wl

⟨Leibniz-Grassmann-Clifford⟩ multilinear differential geometric algebra

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

junyixu/TrixiEnzyme.jl

Autodiff via Enzyme.jl for Trixi.jl

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

alexjc/pytrain

🚃 Automated task/test framework for writing differentiable software.

Language: Python - Size: 243 KB - Last synced at: 7 months ago - Pushed at: almost 5 years ago - Stars: 7 - Forks: 0

wangleiphy/TRG

Differentiable Tensor Renormalization Group for square Ising

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

belkakari/cellular-automata-pytorch

A reproduction and tweaking of Growing Neural Cellular Automata

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

astro-informatics/s2ball

Differentiable and accelerated wavelet transform on the ball with JAX

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

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: 12 months ago - Pushed at: 12 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: about 1 year ago - Pushed at: about 1 year 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: 12 months ago - Pushed at: 12 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: 8 months ago - Pushed at: over 1 year ago - Stars: 37 - Forks: 4

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: about 1 year 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: 8 months ago - Pushed at: almost 4 years ago - Stars: 11 - Forks: 2

FluxML/DiffImages.jl

Differentiable Computer Vision using pure Julia

Language: Julia - Size: 10.5 MB - Last synced at: 6 months ago - Pushed at: almost 2 years 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: 4 months ago - Pushed at: almost 3 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: over 1 year ago - Pushed at: over 1 year ago - Stars: 61 - Forks: 3

ZikangXiong/diff-spec

Differentiable Symbolic Specification

Language: Python - Size: 51.8 KB - Last synced at: over 1 year ago - Pushed at: over 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: over 1 year ago - Pushed at: over 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: over 1 year ago - Pushed at: over 1 year ago - Stars: 329 - Forks: 30

adam-r-kowalski/compute_graph

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

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: about 1 month ago - Pushed at: over 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: over 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: about 1 month 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: almost 2 years ago - Pushed at: over 6 years ago - Stars: 2 - Forks: 0