Jupyter Notebook Jax

Open-source Jupyter Notebook projects categorized as Jax
Edit details
Topics: flax Deep Learning Pytorch Machine Learning Python
Power Real-Time Data Analytics at Scale
Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
www.influxdata.com
featured

Top 15 Jupyter Notebook Jax Projects

  • pyprobml

    Python code for "Probabilistic Machine learning" book by Kevin Murphy

  • whisper-jax

    JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

    • Project mention: [D] What is the most efficient version of OpenAI Whisper? | /r/MachineLearning | 2023-07-12

    Whisper JAX: https://github.com/sanchit-gandhi/whisper-jax. Builds on the Hugging Face implementation. Written in JAX (instead of PyTorch), where you get a 10x or more speed-up if you run it on TPU v4 hardware (I've gotten up to 15x with large batch sizes for super long audio files). Overall, 70-100x faster than OpenAI if you run it on TPU v4

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo

  • neural-tangents

    Fast and Easy Infinite Neural Networks in Python

    • Project mention: Any Deep ReLU Network Is Shallow | news.ycombinator.com | 2023-06-23

    is used to capture the power of a fully-trained deep net of infinite width.

    https://openreview.net/pdf?id=rkl4aESeUH, https://github.com/google/neural-tangents

    > It has long been known that a single-layer fully-connected neural network with an i.i.d. prior over its parameters is equivalent to a Gaussian process (GP), in the limit of infinite network width.

    https://arxiv.org/abs/1711.00165

    And of course, one needs to look back at SVMs applying a kernel function and separating with a line, which looks a lot like an ANN with a single hidden layer followed by a linear mapping.

    https://stats.stackexchange.com/questions/238635/kernel-meth...

  • uvadlc_notebooks

    Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

    • Project mention: I miss the old days where people asked me to recreate “Facebook” or “Twitter” | /r/ProgrammerHumor | 2023-06-04

    So, I don’t have anything simple that’s readily available, and I don’t know how much you’d get from the code itself without some background. But I would recommend the UVA Deep Learning tutorials. Particularly, I’d recommend trying the autoencoder as a good start (tutorial 9). Autoencoders are very easy and fast models to train.

  • brax

    Massively parallel rigidbody physics simulation on accelerator hardware.

  • score_sde

    Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

  • tensor-sensor

    The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo

  • dynamax

    State Space Models library in JAX

  • jax-cfd

    Computational Fluid Dynamics in JAX

    • Project mention: Show HN: WebGPU Particles Simulation | news.ycombinator.com | 2023-12-17

  • jaxrl

    JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

    • Project mention: JAX in Reinforcement Learning | /r/reinforcementlearning | 2023-06-20

    Have you looked at this repo or this repo ?

  • get-started-with-JAX

    The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

    • Project mention: How to stay up-to-date with the latest AI company announcements and events? | /r/singularity | 2023-05-12

  • TF_JAX_tutorials

    All about the fundamental blocks of TF and JAX!

  • indaba-pracs-2022

    Notebooks for the Practicals at the Deep Learning Indaba 2022.

  • hyper-nn

    Easy Hypernetworks in Pytorch and Jax

  • forwardalgorithms

    A tensorflow implementation of the Forward-Forward Algorithm from NeurIPS '22.

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo
NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

Jupyter Notebook Jax related posts

  • Bayesian Analysis with Python

    5 projects | news.ycombinator.com | 10 Feb 2024

  • How to stay up-to-date with the latest AI company announcements and events?

    1 project | /r/singularity | 12 May 2023

  • Physic engine for 3D simulation: which one to use?

    1 project | /r/reinforcementlearning | 8 Oct 2022

  • Tips to become self taught machine learning engineer

    1 project | /r/learnmachinelearning | 4 Aug 2022

  • Jax-CFD: Computational Fluid Dynamics in Jax

    1 project | news.ycombinator.com | 6 May 2022

  • Deep Learning Poised to ‘Blow Up’ Famed Fluid Equations

    4 projects | news.ycombinator.com | 15 Apr 2022

  • AskScience AMA Series: We are seven leading scientists specializing in the intersection of machine learning and neuroscience, and we're working to democratize science education online. Ask Us Anything about computational neuroscience or science education!

    7 projects | /r/askscience | 15 Apr 2022
  • A note from our sponsor - InfluxDB
    www.influxdata.com | 10 May 2024
    Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality. Learn more →

Index

What are some of the best open-source Jax projects in Jupyter Notebook? This list will help you:

Project Stars
1 pyprobml 6,273
2 whisper-jax 4,109
3 neural-tangents 2,225
4 uvadlc_notebooks 2,163
5 brax 2,073
6 score_sde 1,242
7 tensor-sensor 745
8 dynamax 617
9 jax-cfd 627
10 jaxrl 576
11 get-started-with-JAX 563
12 TF_JAX_tutorials 258
13 indaba-pracs-2022 172
14 hyper-nn 91
15 forwardalgorithms 9

Sponsored
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com