Jupyter Notebook Jax

Open-source Jupyter Notebook projects categorized as Jax

Top 12 Jupyter Notebook Jax Projects

  • pyprobml

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

  • neural-tangents

    Fast and Easy Infinite Neural Networks in Python

  • SonarLint

    Clean code begins in your IDE with SonarLint. Up your coding game and discover issues early. SonarLint is a free plugin that helps you find & fix bugs and security issues from the moment you start writing code. Install from your favorite IDE marketplace today.

  • brax

    Massively parallel rigidbody physics simulation on accelerator hardware.

    Project mention: Physic engine for 3D simulation: which one to use? | reddit.com/r/reinforcementlearning | 2022-10-08
  • score_sde

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

    Project mention: Ask HN: How to get back into AI? | news.ycombinator.com | 2022-12-10
  • 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.

  • jaxrl

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

    Project mention: CleanRL now has a DDPG + JAX implementation roughly 2.5-4x faster than DDPG + PyTorch | reddit.com/r/reinforcementlearning | 2022-07-12

    https://github.com/ikostrikov/jaxrl would be another great reference implementation. Probably you want to also checkout the docs for jax, flax, and optax.

  • jax-cfd

    Computational Fluid Dynamics in JAX

    Project mention: Jax-CFD: Computational Fluid Dynamics in Jax | news.ycombinator.com | 2022-05-06
  • InfluxDB

    Build time-series-based applications quickly and at scale.. InfluxDB is the Time Series Platform where developers build real-time applications for analytics, IoT and cloud-native services. Easy to start, it is available in the cloud or on-premises.

  • 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: Tips to become self taught machine learning engineer | reddit.com/r/learnmachinelearning | 2022-08-04

    The AI Epiphany

  • dynamax

    State Space Models library in JAX

    Project mention: Parameter estimation of a linear system | reddit.com/r/ControlTheory | 2023-01-08

    If you can write your system in discrete time (e.g. by using forward Euler) you could use pretty straightforwardly a library like https://github.com/probml/dynamax to estimate your unknown parameters given some measured data.

  • 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.

    Project mention: From Deep Learning Foundations to Stable Diffusion | news.ycombinator.com | 2022-09-16

    This year's Deep Learning Indaba had a tutorial on diffusion models in Jax: https://github.com/deep-learning-indaba/indaba-pracs-2022/tr...

  • hyper-nn

    Easy Hypernetworks in Pytorch and Jax

    Project mention: [P] hyper-nn: Easy Hypernetworks in Pytorch and Jax | reddit.com/r/MachineLearning | 2022-08-11

    I’d like to share a project that I’ve been working on called hyper-nn, which gives users the ability to create easily customizable Hypernetworks for almost any torch.nn.Module and flax.linen.Module.

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). The latest post mention was on 2023-01-08.

Jupyter Notebook Jax related posts


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

Project Stars
1 pyprobml 5,394
2 neural-tangents 1,933
3 brax 1,516
4 score_sde 768
5 tensor-sensor 708
6 jaxrl 414
7 jax-cfd 391
8 get-started-with-JAX 382
9 dynamax 331
10 TF_JAX_tutorials 236
11 indaba-pracs-2022 156
12 hyper-nn 65
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives