Awesome Open Source

Programming Languages

Search results for python jax

299 search results found

Jax_metrics ⭐ 19

A metrics library for the JAX ecosystem

Fine-grained, dynamic control of neural network topology in JAX.

Generative neural networks for 3D terrain.

[ICLR 2024] The official implementation of "Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Model"

Jax_pep503 ⭐ 19

PEP 503 repository index for jax[cuda]

Lecture Jax ⭐ 18

Lectures on Quantitative Economics Using JAX

Nixos Nvidia Cuda Python Docker Compose ⭐ 18

A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host

Chronumental ⭐ 18

Estimating time trees from very large phylogenies

Haiku Mup ⭐ 18

A port of muP to JAX/Haiku

Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.

FID computation in Jax/Flax.

Mtj Softtuner ⭐ 18

Create soft prompts for fairseq 13B dense, GPT-J-6B and GPT-Neo-2.7B for free in a Google Colab TPU instance

Federated posterior averaging implemented in JAX

Jaxdecomp ⭐ 18

JAX bindings for the NVIDIA cuDecomp library

Pure static Lie groups in Numpy, Jax, and C++

Efficiently Composable Data Augmentation on the GPU with Jax

Bart Base Jax ⭐ 18

JAX implementation of the bart-base model

Kauldron ⭐ 17

Modular, scalable codebase to train ML models

Automatic Functional Differentiation in JAX

A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.

Popjaxrl ⭐ 16

Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]

JAX implementation of instant-ngp (NeRF part)

Jax Experimental ⭐ 16

JAX for Graphcore IPU (experimental)

Fully JAX-compatible python implementation of the DC-EGM algorithm from Iskhakov, Jorgensen, Rust, and Schjerning (QE, 2017).

A stateful pytree library for training neural networks.

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

A collection of inverse design challenges

A Fast Modular Synthesizer in JAX

A Simple Statistical Distribution Library in JAX

Scimlbenchmarksoutput ⭐ 13

SciML-Bench Benchmarks for Scientific Machine Learning (SciML), Physics-Informed Machine Learning (PIML), and Scientific AI Performance

Whisper Finetune ⭐ 13

Fine-tune and evaluate Whisper models for Automatic Speech Recognition (ASR) on custom datasets or datasets from huggingface.

Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`

Phantom Tensors ⭐ 13

Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing

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

Sphericalharmonics ⭐ 12

Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax

Dataset_grouper ⭐ 12

Gans Jax ⭐ 12

Implementation of several Generative Adversarial Networks in JAX / Flax

A simple Python package for deep learning using forward automatic differentiation based on JAX.

Map single-cell transcriptomes to copy number evolutionary trees.

Nvidia Gpu Ml Library Test ⭐ 11

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being properly picked up

Deep exponential families for single-cell data.

Jaxlinop ⭐ 11

Linear Operators in JAX.

Software to perform multi-ancestry SNP fine-mapping on molecular data

Fjformer ⭐ 10

Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights!

Probabilistic deep learning using JAX

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!

Code repository for the 2023 MICCAI Paper "Differentiable Beamforming for Ultrasound Autofocusing"

Tessellate Ipu ⭐ 10

TessellateIPU: low level Poplar tile programming from Python

tools for graph-based machine-learning potentials in jax

Jax_trainer ⭐ 10

Lightning-like training API for JAX with Flax

Jax So Vits Svc 5.0 ⭐ 10

A so-vits-svc-5.0 rewritten in jax and flax. Works properly on TPU v2/v3.

Roberta Base Mr ⭐ 9

RoBERTa Marathi Language model trained from scratch during huggingface 🤗 x flax community week

Machine learning Calabi-Yau metrics with JAX

Reinforcement learning in pure JAX.

Holographic Reduced Representations ⭐ 9

Holographic Reduced Representations

Higher-Order Interactions

Adahessianjax ⭐ 9

Jax implementation of the AdaHessian optimizer

Tensorwaves ⭐ 9

Python fitter package for multiple computational back-ends

Interoperating between (Deep) Reiforcement Learning libraries

Nablafuzz ⭐ 9

Fuzzing Automatic Differentiation in Deep-Learning Libraries (ICSE'23)

Differentiable and accelerated wavelet transform on the sphere with JAX

Stochastic ⭐ 8

Pychastic is a stochastic differential equations integrator written entirely in python.

Jochastic ⭐ 8

A JAX implementation of stochastic addition.

Functional Diffusion Processes ⭐ 8

Official code for Continuous-Time Functional Diffusion Processes (NeurIPS 2023).

Rl Basics ⭐ 8

Simple single file implementation of Deep Reinforcement Learning algorithms

Polyglot Jax Inference ⭐ 8

TPU에서 한국어용 LLM 추론을 위한 Jax/Flax 구현체입니다.

JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer

Gravitational Wave Analysis using Auto Differentiation with JAX

SymDer: Symbolic Derivative Approach to Discovering Sparse Interpretable Dynamics from Partial Observations

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.

An unofficial jax/haiku implementation of Crystal Graph Convolutional Neural Networks (CGCNN)

Relaxed Machines ⭐ 7

program synthesis with neuro-symbolic differentiable interpreters

Minimalistic framework to run machine learning experiments.

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

A collection of matrix games in JAX

Sb3 Jax Haiku ⭐ 7

stable-baselines with JAX & Haiku

Diffeqzoo ⭐ 7

A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiological models, IVPs, BVPs, and more.

Siren Jax ⭐ 7

Unofficial implementation of Siren with Jax for image representation.

make your statistical research faster

Jaxdecompiler ⭐ 6

Jaxampler ⭐ 6

🐙 An open-source JAX-based statistical sampling toolkit 🧪

Jax Vqvae Gpt ⭐ 6

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

Doc Builder ⭐ 6

Simple Documentation Builder for Ivy Projects.

Ofdft_normalizing Flows ⭐ 6

Nomalizing flows for orbita-free DFT

Starcoder Jax ⭐ 6

a Jax/Flax inference code of StarCoder

Dash Docset Jax ⭐ 6

Create nice dash/zeal docset for JAX+Flax

Jax Spectral ⭐ 5

Short-time Fourier transform (STFT) for JAX

Graph Convolutional Networks in JAX

Netket_fidelity ⭐ 5

Projected time-dependent Variational Monte Carlo (p-tVMC) method based on infidelity optimization for variational simulation of quantum dynamics.

Jax Nn Template ⭐ 5

Deep Learning Template with JAX

Library for scaling deep learning experiments in JAX

Jax_examples ⭐ 5

Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)

Segnn Jax ⭐ 5

Steerable E(3) GNN in jax

Jax wrapper for autograd-differentiable functions

Quantum Orthogonal Nn ⭐ 5

JAX implementation of Classical and Quantum Algorithms for Orthogonal Neural Networks by (Kerenidis et al., 2021)

Jax_tpu_docker ⭐ 5

Provides a Docker image for running JAX on TPU VMs. This repository contains the necessary files to build the Docker image and run JAX on TPUs with ease, making it easy for researchers to take advantage of the power of TPUs for their machine learning projects.

Differentiable Material Point Method

Numerical Continuation using just the residual. 5k+ downloads

An open-source JAX-based gravitational-wave population inference

Differentiable_t1d ⭐ 5

Differentiable simulators for Type 1 Diabetes research in Jax and PyTorch

Related Searches

Python Machine Learning (20,195)

Python Dataset (14,792)

Python Tensorflow (13,736)

Python Deep Learning (13,092)

Python Jupyter Notebook (12,976)

Python Network (11,495)

Python Video Game (10,593)

Python Algorithms (10,033)

Python Natural Language Processing (9,064)

Python Artificial Intelligence (8,580)

201-299 of 299 search results

Privacy | About | Terms | Follow Us On Twitter

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