Awesome Open Source

Programming Languages

Search results for artificial intelligence jax

artificial-intelligence x

11 search results found

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.

Thinc ⭐ 2,774

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

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

Training Operator ⭐ 1,447

Distributed ML Training and Fine-Tuning on Kubernetes

Fortuna ⭐ 824

A Library for Uncertainty Quantification.

Dreamerv3 ⭐ 749

Mastering Diverse Domains through World Models

Pdebench ⭐ 522

PDEBench: An Extensive Benchmark for Scientific Machine Learning

Scimlbenchmarks.jl ⭐ 279

Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solvers. Covers Julia, Python (PyTorch, Jax), MATLAB, R

Squirrel Core ⭐ 271

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

🎲 Vectorized RL game environments in JAX [NeurIPS23]

Flash Attention Jax ⭐ 123

Implementation of Flash Attention in Jax

Stylegan2 Flax Tpu ⭐ 123

🖼 Training StyleGAN2 on TPUs in JAX

Jax Models ⭐ 82

Unofficial JAX implementations of deep learning research papers

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

Mlp Gpt Jax ⭐ 42

A GPT, made only of MLPs, in Jax

Spyx: Spiking Neural Networks in JAX

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

Tensorshare ⭐ 25

🤝 Trade any tensors over the network

Generative neural networks for 3D terrain.

Jax Experimental ⭐ 16

JAX for Graphcore IPU (experimental)

A collection of inverse design challenges

Tessellate Ipu ⭐ 10

TessellateIPU: low level Poplar tile programming from Python

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

Related Searches

Python Artificial Intelligence (7,915)

Machine Learning Artificial Intelligence (4,499)

Deep Learning Artificial Intelligence (2,745)

Jupyter Notebook Artificial Intelligence (2,497)

Artificial Intelligence Neural Network (1,360)

Java Artificial Intelligence (1,340)

Artificial Intelligence Tensorflow (1,225)

Artificial Intelligence Chatgpt (1,141)

Artificial Intelligence Computer Vision (1,009)

Artificial Intelligence Openai (997)

1-11 of 11 search results

Privacy | About | Terms | Follow Us On Twitter

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