Owner

patrick-kidger

7 indexed repositories · View on GitHub

equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Python · 2973 stars
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Python · 2105 stars
jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Python · 1873 stars
optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

Python · 623 stars
typst_pyimage
Typst extension, adding support for generating figures using inline Python code

Python · 119 stars
typsy
Classes/structs, pattern matching, safe counters... and more! Your one-stop library for programming tools not already in core Typst.

Typst · 30 stars
patstdlib
Some standard pieces I find useful in Typst: fonts, subfigures, algorithms, labeled enums, etc.

Typst · 13 stars

Beginner friendly Everything

Open All

Question: Solving an SDE by applying an ODE Runge-Kutta method to the Stratonovitch form in Diffrax Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/diffrax#769 · 1 comment · Opened Sep 14, 2026
enable-referable-subfigures`: cross-references to a subfigure's `ref` supplement hardcode "Panel" and omit the parent figure number Open

Difficulty 3/5 1-2 days Newbie friendliness 68/100

patrick-kidger/patstdlib#1 · 1 comment · Opened Sep 11, 2026
`error_if` allocates its error branch from a shape, dropping custom array types (quax) Open

Difficulty 4/5 3-5 days Newbie friendliness 68/100

patrick-kidger/equinox#1257 · Opened Aug 19, 2026
`AbstractVar` is indistinguishable from a real field to type checkers, so it is counted in the synthesised `__init__` Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1256 · 1 comment · Opened Aug 18, 2026
Zero-order-hold inputs with fixed-step solvers in a single diffeqsolve Open

Difficulty 4/5 3-5 days Newbie friendliness 52/100

patrick-kidger/diffrax#766 · 4 comments · Opened Aug 13, 2026
problems when calling diffeqsolve inside a blackjax sampler with *shard_map*. Open

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/diffrax#765 · 6 comments · Opened Aug 10, 2026
`GoldenSearch` returns wrong results for any bracket not straddling zero Open

Difficulty 1/5 Under an hour Newbie friendliness 82/100

patrick-kidger/optimistix#236 · 1 comment · Opened Aug 7, 2026
Serialization of rng keys Open

Difficulty 3/5 1-2 days Newbie friendliness 62/100

patrick-kidger/equinox#1255 · 4 comments · Opened Aug 3, 2026
FR: Allow named args to be optional. Open

Difficulty 3/5 1-2 days Newbie friendliness 55/100

patrick-kidger/typsy#2 · Opened Jul 25, 2026
Equinox compatibility with external `PyTreeNode` Open

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/equinox#1248 · 3 comments · Opened Jun 26, 2026
Instantiating Module inherited from abc.ABC returns __class__ assignment error on older versions Open

Difficulty 3/5 1-2 days Newbie friendliness 52/100

patrick-kidger/equinox#1246 · Opened Jun 22, 2026
How to sample fresh noise at each step for a non-linear noise-perturbed ODE ``dX = f(X, z)dt``? Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/diffrax#759 · 8 comments · Opened Jun 10, 2026
Documentation and implementation of ConstantStepSize do not quite align Open

Difficulty 3/5 1-2 days Newbie friendliness 52/100

patrick-kidger/diffrax#757 · Opened May 27, 2026
Best practices concerning referential transparency in `equinox` Open

question

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1236 · 1 comment · Opened May 20, 2026
`equinox.error_if` hangs on error if one of the arguments has a multi-device replicated sharding Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/equinox#1232 · 2 comments · Opened May 19, 2026
Example of how to wrap a concrete subclass within the abstract/final design pattern Open

question

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/equinox#1231 · 5 comments · Opened May 18, 2026
Fixed point tangent solve Open

question

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/optimistix#231 · 1 comment · Opened May 16, 2026
NaNs encountered when taking the quotient of two arrays Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/optimistix#229 · 3 comments · Opened May 1, 2026
Enforcing Absolute Unitarity in Differentiable Physics: The Float-to-Integer Ledger Approach Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

patrick-kidger/diffrax#750 · Opened Apr 30, 2026
Attaching parameter metadata (bounds, priors) at instantiation using "make_with_meta" or "make_shadow" Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1224 · 5 comments · Opened Apr 22, 2026
Leave some elements of `jax` array unchanged during `diffeqsolve` without computation Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#747 · 1 comment · Opened Apr 16, 2026
FYI: Jax v0.9.2 made some nice fixes to TypedNdArray Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/equinox#1206 · 3 comments · Opened Apr 7, 2026
Is there a preferred way to compute the jacobian of a non-holomorphic function in equinox? Open

question

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1201 · 1 comment · Opened Mar 24, 2026
Low Storage Methods Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#743 · 3 comments · Opened Mar 17, 2026
wrapped_fn recompiles and hits torch._dynamo config.recompile_limit Open

question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/jaxtyping#384 · 2 comments · Opened Mar 16, 2026
Return value of the vector field at t0 without recomputation Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#741 · 2 comments · Opened Mar 13, 2026
Self-Scaled Broyden Family of Quasi-Newton Methods implementation Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/optimistix#223 · 1 comment · Opened Mar 12, 2026
Status of constrained non-linear optimization Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/optimistix#222 · 5 comments · Opened Mar 9, 2026
nn.LayerNorm two-pass variance produces ~27% more HLO reduce ops under vmap + grad Open

Difficulty 2/5 1-3 hours Newbie friendliness 52/100

patrick-kidger/equinox#1196 · 4 comments · 1 reaction · Opened Mar 8, 2026
nn.Linear operand order causes unnecessary permutes under vmap + grad Open

Difficulty 2/5 1-3 hours Newbie friendliness 76/100

patrick-kidger/equinox#1195 · 4 comments · 1 reaction · Opened Mar 8, 2026
Class-level scoping question Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#383 · 1 comment · 1 reaction · Opened Mar 4, 2026
Running diffeqsolve for a fixed number of *accepted* steps Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#740 · 1 comment · Opened Mar 4, 2026
Support for Sharding 2.0 2.0 Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/diffrax#738 · 2 comments · Opened Feb 26, 2026
test_jax_transform_warn fails with jax 0.9 Open

Difficulty 3/5 1-2 days Newbie friendliness 38/100

patrick-kidger/equinox#1186 · 3 comments · Opened Feb 17, 2026
Custom error classes with `error_if` Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/equinox#1185 · 1 comment · Opened Feb 17, 2026
Gradients of state variables in post-event simulations Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#729 · 15 comments · Opened Feb 11, 2026
[Feature Request] `+` to match "at least one"/"1 or more" array axes Open

Difficulty 4/5 3-5 days Newbie friendliness 42/100

patrick-kidger/jaxtyping#378 · Opened Feb 3, 2026
Is it practical to use axis names in runtime functionality? Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/jaxtyping#377 · 2 comments · Opened Feb 3, 2026
How to diagnose/profile performance issues Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/equinox#1177 · Opened Feb 2, 2026
dataset.map(jaxtyped_function, fn_kwargs={"model": model}) will be very slow. Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#375 · 1 comment · Opened Jan 29, 2026
Tolerance schemes when using iterative solvers in least squares Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/optimistix#208 · 6 comments · Opened Jan 29, 2026
Correct use of custom `filter_spec` Open

question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/equinox#1176 · 2 comments · Opened Jan 28, 2026
Is `throw=False` option in `root_find` meant to catch solver fails? Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/optimistix#206 · 1 comment · Opened Jan 25, 2026
`test_sharding_*` tests are failing with jax 0.9.0 Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

patrick-kidger/equinox#1171 · 4 comments · Opened Jan 24, 2026
Creating Support for Delay Differential Equations Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/diffrax#721 · 3 comments · Opened Jan 20, 2026
XLA Convolution algorithms fail on GPU Open

question

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/equinox#1166 · 7 comments · Opened Jan 13, 2026
auto-parallelization alongside `vmap` is very slow Open

question

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/diffrax#718 · 4 comments · Opened Jan 12, 2026
Kvaerno, KenCarp solvers fail jax tracing for a multicomponent state and jax 0.8.2 Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/diffrax#717 · 1 comment · Opened Jan 12, 2026
A question about optx.root_find() to find roots Open

question

Difficulty 3/5 1-2 days Newbie friendliness 25/100

patrick-kidger/optimistix#204 · 2 comments · Opened Jan 8, 2026
Scalable optimization for large data Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/optimistix#201 · 3 comments · Opened Dec 30, 2025

Showing the newest 100

This page lists what was indexed most recently. Advanced filter has the whole inventory, narrowed by language, difficulty and how long a task takes.

Open advanced filter

patrick-kidger

Showing the newest 100

Get new issues in your inbox