Awesome Open Source

Programming Languages

The Top 3 Python Jax Xla Open Source Projects

Open source projects categorized as Python Jax Xla

Categories > Jax

Categories > Programming Languages > Python

Categories > Xla

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.

most recent commit 5 months ago

Extending Jax ⭐ 293

Extending JAX with custom C++ and CUDA code

most recent commit 6 months ago

Jaxdecomp ⭐ 18

JAX bindings for the NVIDIA cuDecomp library

most recent commit 3 months ago

Categories

Application Programming Interfaces

Artificial Intelligence

Cloud Computing

Command Line Interface

Computer Science

Configuration Management

Content Management

Data Processing

Integrated Development Environments

Learning Resources

Lists Of Projects

Machine Learning

Operating Systems

Package Managers

Programming Languages

Runtime Environments

Software Architecture

Software Development

Software Performance

Software Quality

Text Processing

User Interface Components

Version Control

Web User Interface

Privacy | About | Terms | Follow Us On Twitter

Downloads, Dependent Repos, Dependent Packages, Total Releases, Latest Releases data powered by Libraries.io.

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