jax
Here are 970 public repositories matching this topic...
Common practices for distributed training using various backends
-
Updated
Sep 23, 2023 - Python
Code for various probabilistic deep learning models
-
Updated
Jun 28, 2023 - Jupyter Notebook
Conversion of LAST library from JAX to PyTorch
-
Updated
Dec 14, 2023 - Python
A setup for solving shape optimization problems depending on a partial differential equation using boundary integral methods and the Python library JAX. The code was developed as part of a Bachelor thesis project in Engineering Mathematics at the Royal Institute of Technology in Spring 2024 by Rebecka Johansson and Asta Stensson.
-
Updated
Jun 10, 2024 - Jupyter Notebook
Estêvão's Single Stage Optimizer of alpha particles via differentiable JAX code
-
Updated
Jul 5, 2024 - Python
A simple diffusion (generative) library I made as part of my diffusion experiments, in flax and jax. Trying my best to implement most of the latest research methodologies
-
Updated
Jul 5, 2024 - Jupyter Notebook
Toy implementations of popular deep learning optimizers from scratch in JAX
-
Updated
Jun 20, 2021 - Jupyter Notebook
This is an implementation of linear regression using JAX and a functional programming approach. You can find a related blog post at the following link: https://medium.com/@sahinadirhan/simple-linear-regression-using-jax-5ef2eefb8cf4
-
Updated
Jul 17, 2021 - Jupyter Notebook
A JAX implementation of DEformer for arbitrary conditioning.
-
Updated
May 29, 2022 - Python
-
Updated
Mar 19, 2023 - Jupyter Notebook
Machine learning samples
-
Updated
Mar 21, 2024
This tutorial is part of the "Intensive Course Machine Learning: From Basics to Advanced Concepts" of the QuCoLiMa research network and the International Max Planck Research School Physics of Light.
-
Updated
May 7, 2024 - Jupyter Notebook
A generalized implementation of Grad-CAM for Flax
-
Updated
Oct 23, 2023 - Python
A comprehensive programming environment designed to facilitate research and development at the intersection of high-energy physics and machine learning.
-
Updated
May 13, 2024 - Dockerfile
minimal forward-mode automatic differentiation using python's abstract syntax tree
-
Updated
Jun 28, 2024 - Python
Python code for "Probabilistic Machine learning" book by Kevin Murphy
-
Updated
Feb 11, 2023 - Jupyter Notebook
Searching for galaxy satellites by their impact on lens potential in strong gravitational lensing setup
-
Updated
Sep 12, 2021 - Jupyter Notebook
Improve this page
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."