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. Learn more →
Top 12 Jupyter Notebook Jax Projects
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Fast and Easy Infinite Neural Networks in Python
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.
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
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
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
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.
Computational Fluid Dynamics in JAXProject mention: Jax-CFD: Computational Fluid Dynamics in Jax | news.ycombinator.com | 2022-05-06
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.
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
State Space Models library in JAXProject 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.
All about the fundamental blocks of TF and JAX!
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...
Easy Hypernetworks in Pytorch and JaxProject 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.
Jupyter Notebook Jax related posts
Physic engine for 3D simulation: which one to use?
1 project | reddit.com/r/reinforcementlearning | 8 Oct 2022
Tips to become self taught machine learning engineer
1 project | reddit.com/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 | reddit.com/r/askscience | 15 Apr 2022
Deep learning for robotics
3 projects | reddit.com/r/buildapc | 4 Apr 2022
[D] Advice on Hardware Setup for Robotics
3 projects | reddit.com/r/MachineLearning | 4 Apr 2022
A note from our sponsor - SonarLint
www.sonarlint.org | 27 Jan 2023
What are some of the best open-source Jax projects in Jupyter Notebook? This list will help you: