jax
Here are 952 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
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
minimal forward-mode automatic differentiation using python's abstract syntax tree
-
Updated
May 10, 2024 - 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
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
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
This is the official repository of the paper "RoCourseNet: Distributionally Robust Training of a Prediction Aware Recourse Model".
-
Updated
Oct 22, 2023 - 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."