Awesome Open Source

Programming Languages

Search results for jupyter notebook jax

jupyter-notebook x

66 search results found

Pyprobml ⭐ 6,066

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

Whisper Jax ⭐ 3,824

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

Neural Tangents ⭐ 2,149

Fast and Easy Infinite Neural Networks in Python

Massively parallel rigidbody physics simulation on accelerator hardware.

Uvadlc_notebooks ⭐ 1,860

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

Machine_learning_refined ⭐ 1,574

Notes, examples, and Python demos for the 2nd edition of the textbook "Machine Learning Refined" (published by Cambridge University Press).

Score_sde ⭐ 782

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

Tensor Sensor ⭐ 634

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

Treasure Of Transformers ⭐ 541

💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs, official repo along with colab Notebooks. 🛫☑️

Dynamax ⭐ 540

State Space Models library in JAX

Jax Cfd ⭐ 527

Computational Fluid Dynamics in JAX

Get Started With Jax ⭐ 487

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.

Learning Python Physics Informed Machine Learning Pinns Deeponets ⭐ 297

Physics Informed Machine Learning Tutorials (Pytorch and Jax)

Commonlooputils ⭐ 261

CLU lets you write beautiful training loops in JAX.

Additive manufacturing simulation with JAX.

Indaba Pracs 2022 ⭐ 151

Notebooks for the Practicals at the Deep Learning Indaba 2022.

Tf_jax_tutorials ⭐ 135

All about the fundamental blocks of TF and JAX!

Robustness Vit ⭐ 97

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

A Python library for integrating model-based and judgmental forecasting

A multi-modal CLIP model trained on the medical dataset ROCO

Jax Fenics Adjoint ⭐ 71

Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint

Cr Sparse ⭐ 70

Functional models and algorithms for sparse signal processing

Hyper Nn ⭐ 65

Easy Hypernetworks in Pytorch and Jax

SciFive: a text-text transformer model for biomedical literature

Alphatrade ⭐ 55

JAX-LOB: A GPU-Accelerated limit order book simulator to unlock large scale reinforcement learning for trading

Revisiting_rainbow ⭐ 45

Revisiting Rainbow

Pytorch and Jax code for the Madam optimiser.

S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.

Scalable Optical Flow-based Image Montaging and Alignment

Rieoptax ⭐ 36

Riemannian Optimization Using JAX

Spyx: Spiking Neural Networks in JAX

Physics Informed Neural Network In Jax ⭐ 31

Example problems in Physics informed neural network in JAX

Tigercontrol ⭐ 31

Google AI Princeton control framework

Vit Jax2tf ⭐ 26

This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.

Jax Dataloader ⭐ 20

Pytorch-like dataloaders in JAX.

Big_vision_experiments ⭐ 19

Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.

Federated posterior averaging implemented in JAX

Point Cloud Galaxy Diffusion ⭐ 15

Dffusion modeling for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141

Cosmopower Jax ⭐ 13

Differentiable cosmological emulators: the JAX version of cosmopower

Parallel Non Linear Gaussian Smoothers ⭐ 13

Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.

A lightweight didactic library of kernel methods using the back-end JAX.

Deep Learning Projects ⭐ 13

Best Deep Learning Projects for Advanced Learners

Jaxrenderer ⭐ 13

Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrenderer/wiki; PR: https://github.com/google/brax/pull/367

Videomae ⭐ 12

[NeurIPS'22] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training

An efficient method for the conversion from internal to Cartesian coordinates that utilizes the platform-agnostic JAX Python library.

Detect Fake Text ⭐ 12

LLM - Detect AI Generated Text || Identify which essay was written by a large language model

Gans Jax ⭐ 12

Implementation of several Generative Adversarial Networks in JAX / Flax

A JAX Implementation of the Twin Delayed DDPG Algorithm

Energy Transformer Graph ⭐ 11

This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classification

Bit Jax2tf ⭐ 11

This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.

Yerbamate ⭐ 10

A framework-agnostic deep learning package and experiment manager

Flax_nerf ⭐ 9

Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API

Mlpmixer Jax2tf ⭐ 9

This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.

Ad Kernels ⭐ 9

Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"

Gnn For Combinatorial Optimization ⭐ 8

JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.

Forward Forward Algorithm ⭐ 8

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

A flexible, fast and scalable python library for Self-Organizing Maps

Vit Tf Hub Application ⭐ 6

Build and fine-tune your Image Classifier using a Vision Transformer Model from TensorFlow Hub

Differentiable microlensing

Conservation Laws Manifold Learning ⭐ 6

Discovering Conservation Laws using Optimal Transport and Manifold Learning

Machine learning pipeline trained to detect X-ray cavities in Chandra images of early-type galaxies.

Learningjax ⭐ 5

Learning Notes and resources on Jax

Iterative and Parametric Gaussianization with JAX.

Gsoc 22 Tensorflow Resources And Notebooks ⭐ 5

GSoC'22 @ TensorFlow Notebooks, Code and much more

Related Searches

Python Jupyter Notebook (12,976)

Jupyter Notebook Machine Learning (8,463)

Jupyter Notebook Dataset (6,824)

Jupyter Notebook Deep Learning (6,566)

Jupyter Notebook Tensorflow (4,771)

Jupyter Notebook Data Science (4,256)

Jupyter Notebook Convolutional Neural Networks (4,218)

Jupyter Notebook Classification (3,939)

Jupyter Notebook Neural (3,926)

Jupyter Notebook Pytorch (3,877)

1-66 of 66 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.