Skip to content
View patrick-kidger's full-sized avatar
1.7k followers · 2 following

Sponsors

@lucidrains
@Artur-Galstyan
@dlwh
@arf-themascoteers
@sequoiar
@taskswithcode

Achievements

Achievement: Galaxy Brainx2Achievement: Pair ExtraordinaireAchievement: QuickdrawAchievement: YOLOAchievement: Starstruckx3Achievement: Pull Sharkx3Achievement: Arctic Code Vault Contributor
BetaSend feedback

Achievements

Achievement: Galaxy Brainx2Achievement: Pair ExtraordinaireAchievement: QuickdrawAchievement: YOLOAchievement: Starstruckx3Achievement: Pull Sharkx3Achievement: Arctic Code Vault Contributor
BetaSend feedback
Block or Report

Block or report patrick-kidger

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Add an optional note:
Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
patrick-kidger/README.md

JAX scientific ML ecosystem:

Probably the reason you're here. I would highlight:

  1. Equinox: elegant neural networks. GitHub Repo stars

  2. Diffrax: numerical ODE/SDE solvers. GitHub Repo stars

  3. jaxtyping: shape/dtype annotations for arrays. GitHub Repo stars

  4. Lineax: linear/least-squares solvers. GitHub Repo stars (new!)

  5. Optimistix: root finding, least squares, etc. GitHub Repo stars (new!)

Other links:

Me: currently doing bioML at Google X. Previously PhD in mathematics from Oxford + The Alan Turing Institute, on neural ODEs/SDEs/CDEs.

Pinned

  1. google-research/torchsde google-research/torchsde Public

    Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

    Python 1.4k 177

  2. torchtyping torchtyping Public

    Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

    Python 1.3k 31

  3. equinox equinox Public

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

    Python 1.6k 102

  4. diffrax diffrax Public

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

    Python 1.1k 91

  5. google/jaxtyping google/jaxtyping Public

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

    Python 703 30

  6. NeuralCDE NeuralCDE Public

    Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

    Python 564 65

1,637 contributions in the last year

No contributions on December 11th.No contributions on December 18th.2 contributions on December 25th.No contributions on January 1st.No contributions on January 8th.No contributions on January 15th.8 contributions on January 22nd.No contributions on January 29th.No contributions on February 5th.No contributions on February 12th.No contributions on February 19th.8 contributions on February 26th.13 contributions on March 5th.1 contribution on March 12th.7 contributions on March 19th.7 contributions on March 26th.No contributions on April 2nd.No contributions on April 9th.5 contributions on April 16th.4 contributions on April 23rd.9 contributions on April 30th.12 contributions on May 7th.18 contributions on May 14th.5 contributions on May 21st.12 contributions on May 28th.2 contributions on June 4th.2 contributions on June 11th.No contributions on June 18th.7 contributions on June 25th.No contributions on July 2nd.6 contributions on July 9th.4 contributions on July 16th.15 contributions on July 23rd.No contributions on July 30th.No contributions on August 6th.9 contributions on August 13th.2 contributions on August 20th.No contributions on August 27th.No contributions on September 3rd.2 contributions on September 10th.2 contributions on September 17th.No contributions on September 24th.No contributions on October 1st.5 contributions on October 8th.4 contributions on October 15th.8 contributions on October 22nd.No contributions on October 29th.No contributions on November 5th.No contributions on November 12th.No contributions on November 19th.No contributions on November 26th.No contributions on December 3rd.9 contributions on December 10th.No contributions on December 12th.No contributions on December 19th.3 contributions on December 26th.8 contributions on January 2nd.5 contributions on January 9th.No contributions on January 16th.2 contributions on January 23rd.8 contributions on January 30th.No contributions on February 6th.1 contribution on February 13th.17 contributions on February 20th.3 contributions on February 27th.9 contributions on March 6th.9 contributions on March 13th.9 contributions on March 20th.7 contributions on March 27th.2 contributions on April 3rd.No contributions on April 10th.2 contributions on April 17th.No contributions on April 24th.14 contributions on May 1st.8 contributions on May 8th.11 contributions on May 15th.4 contributions on May 22nd.10 contributions on May 29th.2 contributions on June 5th.2 contributions on June 12th.1 contribution on June 19th.1 contribution on June 26th.2 contributions on July 3rd.10 contributions on July 10th.6 contributions on July 17th.9 contributions on July 24th.2 contributions on July 31st.No contributions on August 7th.2 contributions on August 14th.4 contributions on August 21st.3 contributions on August 28th.2 contributions on September 4th.16 contributions on September 11th.3 contributions on September 18th.13 contributions on September 25th.17 contributions on October 2nd.11 contributions on October 9th.13 contributions on October 16th.9 contributions on October 23rd.2 contributions on October 30th.2 contributions on November 6th.9 contributions on November 13th.No contributions on November 20th.3 contributions on November 27th.2 contributions on December 4th.1 contribution on December 11th.No contributions on December 13th.No contributions on December 20th.13 contributions on December 27th.1 contribution on January 3rd.2 contributions on January 10th.No contributions on January 17th.9 contributions on January 24th.No contributions on January 31st.No contributions on February 7th.5 contributions on February 14th.15 contributions on February 21st.4 contributions on February 28th.22 contributions on March 7th.4 contributions on March 14th.1 contribution on March 21st.6 contributions on March 28th.1 contribution on April 4th.3 contributions on April 11th.6 contributions on April 18th.3 contributions on April 25th.2 contributions on May 2nd.10 contributions on May 9th.3 contributions on May 16th.5 contributions on May 23rd.6 contributions on May 30th.3 contributions on June 6th.1 contribution on June 13th.6 contributions on June 20th.2 contributions on June 27th.8 contributions on July 4th.5 contributions on July 11th.23 contributions on July 18th.7 contributions on July 25th.12 contributions on August 1st.4 contributions on August 8th.No contributions on August 15th.No contributions on August 22nd.1 contribution on August 29th.10 contributions on September 5th.9 contributions on September 12th.3 contributions on September 19th.7 contributions on September 26th.12 contributions on October 3rd.13 contributions on October 10th.6 contributions on October 17th.1 contribution on October 24th.1 contribution on October 31st.4 contributions on November 7th.No contributions on November 14th.No contributions on November 21st.No contributions on November 28th.2 contributions on December 5th.No contributions on December 14th.1 contribution on December 21st.4 contributions on December 28th.1 contribution on January 4th.3 contributions on January 11th.No contributions on January 18th.3 contributions on January 25th.No contributions on February 1st.No contributions on February 8th.1 contribution on February 15th.7 contributions on February 22nd.11 contributions on March 1st.38 contributions on March 8th.3 contributions on March 15th.5 contributions on March 22nd.8 contributions on March 29th.5 contributions on April 5th.1 contribution on April 12th.6 contributions on April 19th.2 contributions on April 26th.3 contributions on May 3rd.8 contributions on May 10th.5 contributions on May 17th.No contributions on May 24th.5 contributions on May 31st.23 contributions on June 7th.9 contributions on June 14th.4 contributions on June 21st.No contributions on June 28th.8 contributions on July 5th.8 contributions on July 12th.No contributions on July 19th.7 contributions on July 26th.25 contributions on August 2nd.4 contributions on August 9th.7 contributions on August 16th.No contributions on August 23rd.No contributions on August 30th.No contributions on September 6th.3 contributions on September 13th.10 contributions on September 20th.23 contributions on September 27th.3 contributions on October 4th.1 contribution on October 11th.16 contributions on October 18th.4 contributions on October 25th.3 contributions on November 1st.6 contributions on November 8th.3 contributions on November 15th.3 contributions on November 22nd.No contributions on November 29th.1 contribution on December 6th.1 contribution on December 15th.1 contribution on December 22nd.4 contributions on December 29th.No contributions on January 5th.3 contributions on January 12th.1 contribution on January 19th.9 contributions on January 26th.No contributions on February 2nd.No contributions on February 9th.7 contributions on February 16th.11 contributions on February 23rd.3 contributions on March 2nd.14 contributions on March 9th.2 contributions on March 16th.4 contributions on March 23rd.No contributions on March 30th.No contributions on April 6th.9 contributions on April 13th.5 contributions on April 20th.5 contributions on April 27th.12 contributions on May 4th.4 contributions on May 11th.8 contributions on May 18th.4 contributions on May 25th.15 contributions on June 1st.5 contributions on June 8th.1 contribution on June 15th.6 contributions on June 22nd.2 contributions on June 29th.4 contributions on July 6th.12 contributions on July 13th.6 contributions on July 20th.2 contributions on July 27th.6 contributions on August 3rd.No contributions on August 10th.8 contributions on August 17th.1 contribution on August 24th.11 contributions on August 31st.12 contributions on September 7th.8 contributions on September 14th.5 contributions on September 21st.1 contribution on September 28th.15 contributions on October 5th.2 contributions on October 12th.3 contributions on October 19th.3 contributions on October 26th.11 contributions on November 2nd.No contributions on November 9th.No contributions on November 16th.No contributions on November 23rd.No contributions on November 30th.No contributions on December 7th.No contributions on December 16th.1 contribution on December 23rd.2 contributions on December 30th.1 contribution on January 6th.No contributions on January 13th.5 contributions on January 20th.3 contributions on January 27th.1 contribution on February 3rd.No contributions on February 10th.1 contribution on February 17th.3 contributions on February 24th.2 contributions on March 3rd.3 contributions on March 10th.2 contributions on March 17th.3 contributions on March 24th.No contributions on March 31st.No contributions on April 7th.5 contributions on April 14th.No contributions on April 21st.5 contributions on April 28th.6 contributions on May 5th.7 contributions on May 12th.1 contribution on May 19th.13 contributions on May 26th.4 contributions on June 2nd.7 contributions on June 9th.No contributions on June 16th.No contributions on June 23rd.No contributions on June 30th.5 contributions on July 7th.4 contributions on July 14th.24 contributions on July 21st.7 contributions on July 28th.17 contributions on August 4th.4 contributions on August 11th.4 contributions on August 18th.2 contributions on August 25th.8 contributions on September 1st.14 contributions on September 8th.5 contributions on September 15th.18 contributions on September 22nd.11 contributions on September 29th.20 contributions on October 6th.10 contributions on October 13th.No contributions on October 20th.No contributions on October 27th.2 contributions on November 3rd.4 contributions on November 10th.1 contribution on November 17th.No contributions on November 24th.No contributions on December 1st.No contributions on December 8th.No contributions on December 17th.7 contributions on December 24th.2 contributions on December 31st.No contributions on January 7th.1 contribution on January 14th.6 contributions on January 21st.No contributions on January 28th.No contributions on February 4th.No contributions on February 11th.No contributions on February 18th.8 contributions on February 25th.6 contributions on March 4th.2 contributions on March 11th.No contributions on March 18th.5 contributions on March 25th.No contributions on April 1st.No contributions on April 8th.1 contribution on April 15th.No contributions on April 22nd.No contributions on April 29th.1 contribution on May 6th.6 contributions on May 13th.No contributions on May 20th.14 contributions on May 27th.4 contributions on June 3rd.5 contributions on June 10th.No contributions on June 17th.No contributions on June 24th.No contributions on July 1st.No contributions on July 8th.6 contributions on July 15th.9 contributions on July 22nd.3 contributions on July 29th.4 contributions on August 5th.4 contributions on August 12th.No contributions on August 19th.2 contributions on August 26th.No contributions on September 2nd.1 contribution on September 9th.No contributions on September 16th.1 contribution on September 23rd.1 contribution on September 30th.1 contribution on October 7th.2 contributions on October 14th.14 contributions on October 21st.4 contributions on October 28th.No contributions on November 4th.1 contribution on November 11th.No contributions on November 18th.No contributions on November 25th.4 contributions on December 2nd.12 contributions on December 9th.
Contribution Graph
Day of Week December January February March April May June July August September October November December
Sunday
Monday
Tuesday
Wednesday
Thursday
Friday
Saturday
Learn how we count contributions
Less
No contributions.
Low contributions.
Medium-low contributions.
Medium-high contributions.
High contributions.
More

Contribution activity

December 2023

Created 18 commits in 5 repositories
Opened 8 pull requests in 4 repositories
patrick-kidger/optimistix 1 open 1 closed
google/lineax 1 open 1 closed
patrick-kidger/equinox 2 merged
google/jaxtyping 2 merged
Reviewed 3 pull requests in 2 repositories
patrick-kidger/diffrax 2 pull requests
google/lineax 1 pull request
Opened 1 issue in 1 repository
patrick-kidger/equinox 1 open
1 contribution in private repositories Dec 4

Seeing something unexpected? Take a look at the GitHub profile guide.