Awesome Open Source

Programming Languages

Search results for machine learning jax

machine-learning x

99 search results found

Transformers ⭐ 124,049

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Keras ⭐ 60,949

Deep Learning for humans

D2l En ⭐ 20,613

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

Best Of Ml Python ⭐ 14,990

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

The Unified AI Framework

Wandb ⭐ 8,204

🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.

Trax — Deep Learning with Clear Code and Speed

Pyprobml ⭐ 6,066

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

Datasets ⭐ 4,094

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Interviews.ai ⭐ 3,146

It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.

Training and serving large-scale neural networks with auto parallelization.

Dm Haiku ⭐ 2,807

JAX-based neural network library

Thinc ⭐ 2,774

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

Foolbox ⭐ 2,600

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

Pennylane ⭐ 2,022

PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.

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).

Tensorly ⭐ 1,466

TensorLy: Tensor Learning in Python.

Training Operator ⭐ 1,447

Distributed ML Training and Fine-Tuning on Kubernetes

Jraph ⭐ 1,233

A Graph Neural Network Library in Jax

Transformerengine ⭐ 1,172

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

Awesome Jax ⭐ 1,156

JAX - A curated list of resources https://github.com/google/jax

Jraph ⭐ 1,141

A Graph Neural Network Library in Jax

Diffrax ⭐ 1,131

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Fortuna ⭐ 824

A Library for Uncertainty Quantification.

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

Prompt Tuning ⭐ 580

Original Implementation of Prompt Tuning from Lester, et al, 2021

Pdebench ⭐ 522

PDEBench: An Extensive Benchmark for Scientific Machine Learning

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.

Machine learning algorithms for many-body quantum systems

Neuralcompression ⭐ 430

A collection of tools for neural compression enthusiasts.

Tpu Starter ⭐ 381

Everything you want to know about Google Cloud TPU

Git Re Basin ⭐ 378

Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"

Machine Learning Guide ⭐ 366

Machine learning Guide. Learn all about Machine Learning Tools, Libraries, Frameworks, Large Language Models (LLMs), and Training Models.

Gaussian processes in JAX.

PIX is an image processing library in JAX, for JAX.

Learning Python Physics Informed Machine Learning Pinns Deeponets ⭐ 297

Physics Informed Machine Learning Tutorials (Pytorch and Jax)

Squirrel Core ⭐ 271

A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way 🌰

Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Isaac Orbit and Omniverse Isaac Gym

Commonlooputils ⭐ 261

CLU lets you write beautiful training loops in JAX.

🎲 Vectorized RL game environments in JAX [NeurIPS23]

Tensorcircuit ⭐ 230

Tensor network based quantum software framework for the NISQ era

Functional tensors for probabilistic programming

Jaxfluids ⭐ 203

Differentiable Fluid Dynamics Package

Bayesnewton ⭐ 196

Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.

Easy Neural Ode ⭐ 186

Code for the paper "Learning Differential Equations that are Easy to Solve"

Concise deep learning for JAX

Tf_jax_tutorials ⭐ 135

All about the fundamental blocks of TF and JAX!

Sunstreaker ⭐ 126

以jax为后端的类似keras的框架

Swanlab ⭐ 108

🧐SwanLab: track and visualize all the pieces of your machine learning pipeline. Join our WeChat ⬇️

Autoray ⭐ 106

Abstract your array operations.

Reinforcement learning library in JAX.

Flashbax ⭐ 100

⚡ Flashbax: Accelerated Replay Buffers in JAX

EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥

Jax Models ⭐ 82

Unofficial JAX implementations of deep learning research papers

Jestimator ⭐ 77

Amos optimizer with JEstimator lib.

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

Auto-differentiable and hardware-accelerated force density method

Ngc Learn ⭐ 71

NGC-Learn: Predictive Coding and Neurobiologically-Motivated Learning in Python

3D Vision functions with end-to-end support for deep learning developers, written in Ivy.

Jax Resnet ⭐ 67

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

Hyper Nn ⭐ 65

Easy Hypernetworks in Pytorch and Jax

Reinforced Lib ⭐ 60

Reinforcement learning library

Fully differentiable RL environments, written in Ivy.

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.

Revisiting_rainbow ⭐ 45

Revisiting Rainbow

A Python library for machine-learning and feedback loops on streaming data

Rockpool ⭐ 37

A machine learning library for spiking neural networks. Supports training with both torch and jax pipelines, and deployment to neuromorphic hardware.

Tigercontrol ⭐ 31

Google AI Princeton control framework

🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX

Spyx: Spiking Neural Networks in JAX

Machine Learning eXperiment Utilities

Haiku Geometric ⭐ 26

A collection of graph neural networks implementations in JAX

Corax: Core RL in JAX

Tensorshare ⭐ 25

🤝 Trade any tensors over the network

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

Ml Optimizers Jax ⭐ 23

Toy implementations of some popular ML optimizers using Python/JAX

Susiepca ⭐ 20

Scalable Ultra-Sparse Bayesian PCA

a modular reinforcement learning library with JAX agents

Generative neural networks for 3D terrain.

Haiku Mup ⭐ 18

A port of muP to JAX/Haiku

Jax Experimental ⭐ 16

JAX for Graphcore IPU (experimental)

Causalimages Software ⭐ 15

causalimages: An R package for performing causal inference with image and image sequence data with a JAX backend

Mechanics functions with end-to-end support for deep learning developers, written in Ivy.

Cosmopower Jax ⭐ 13

Differentiable cosmological emulators: the JAX version of cosmopower

Deep Learning Projects ⭐ 13

Best Deep Learning Projects for Advanced Learners

Gans Jax ⭐ 12

Implementation of several Generative Adversarial Networks in JAX / Flax

Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.

Rl_with_resets ⭐ 11

JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"

Yerbamate ⭐ 10

A framework-agnostic deep learning package and experiment manager

Jax Models ⭐ 10

Explore implementations of deep learning concepts like Transformers, Attention, Llama, GPT, InstructGPT, RLHF, Gaussian Processes, Bayesian Inference, Newton Raphson, Distributed Trainers and more!

Jax_trainer ⭐ 10

Lightning-like training API for JAX with Flax

Tessellate Ipu ⭐ 10

TessellateIPU: low level Poplar tile programming from Python

Linkorgs Software ⭐ 9

LinkOrgs: An R package for linking linking records on organizations using half-a-billion open-collaborated records from LinkedIn

cLLM is an Open-source library that use llama-cpp-python and llama.cpp and provide a Low and High level API and allow developer to be more pythonic.

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

Minimalistic framework to run machine learning experiments.

Doc Builder ⭐ 6

Simple Documentation Builder for Ivy Projects.

Jax Vqvae Gpt ⭐ 6

Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.

Conservation Laws Manifold Learning ⭐ 6

Discovering Conservation Laws using Optimal Transport and Manifold Learning

Jax Journey ⭐ 6

A pathway and collection of resources to learning Jax from beginning to advance.

Related Searches

Python Machine Learning (14,099)

Jupyter Notebook Machine Learning (12,247)

Machine Learning Neural Network (4,397)

Machine Learning Tensorflow (4,050)

Machine Learning Natural Language Processing (3,891)

Machine Learning Artificial Intelligence (3,877)

Machine Learning Data Science (3,802)

Machine Learning Pytorch (2,910)

Machine Learning Dataset (2,298)

Machine Learning Computer Vision (1,966)

1-99 of 99 search results

Privacy | About | Terms | Follow Us On Twitter

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