lindseysbrown / Auto-eD
A web based tool for visualization of the forward and reverse modes of automatic differentiation
☆16Updated 4 months ago
Related projects: ⓘ
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆119Updated this week
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆56Updated 2 years ago
- http://cranmer.github.io/stats-ds-book☆65Updated 3 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆87Updated 10 months ago
- Create Numpy .npy files by appending on the growth axis☆59Updated last year
- Stencil computations in JAX☆66Updated 11 months ago
- Datasets and code for results presented in the ProbConserv paper☆51Updated 3 months ago
- Bibtex for various Python science and machine learning software☆31Updated last year
- Datasets and code for results presented in the BOON paper☆41Updated last year
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- A curated list of resources to help with computational research.☆20Updated 2 years ago
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆11Updated 9 months ago
- Solving Optimization Problems with JAX, code and PDF☆15Updated 4 years ago
- Clean up the public namespace of your package!☆51Updated 5 months ago
- Documentation:☆119Updated last year
- The Physics of Energy Based Models☆16Updated 6 months ago
- ☆24Updated last year
- Learning some numerical linear algebra.☆70Updated 3 years ago
- Differentiable interface to FEniCS for JAX☆49Updated 3 years ago
- Running Jax in PyTorch Lightning☆76Updated last year
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- ☆28Updated this week
- ☆17Updated 11 months ago
- Code for the book "The Elements of Differentiable Programming".☆57Updated last month
- Python Algorithms for Randomized Linear Algebra☆46Updated last year
- ☆103Updated last month
- Worked example of the process from Python source to CUDA kernel execution with Numba☆36Updated last week
- Tools for JAX☆44Updated this week
- Orthogonal polynomials in all shapes and sizes.☆182Updated 7 months ago
- A Pytree Module system for Deep Learning in JAX☆215Updated last year