Awesome Open Source

Programming Languages

Search results for jax xla

6 search results found

Awesome Jax ⭐ 1,156

JAX - A curated list of resources https://github.com/google/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.

Mpi4jax ⭐ 348

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

Extending Jax ⭐ 293

Extending JAX with custom C++ and CUDA code

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

Jaxdecomp ⭐ 18

JAX bindings for the NVIDIA cuDecomp library

Related Searches

Python Jax (191)

Deep Learning Jax (80)

Machine Learning Jax (71)

Jupyter Notebook Jax (42)

Tensorflow Jax (41)

Python Xla (13)

Tensorflow Xla (9)

Deep Learning Xla (7)

1-6 of 6 search results

Privacy | About | Terms | Follow Us On Twitter

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